Saya ingin membuat scatterplot (menggunakan matplotlib) di mana titik-titik diarsir menurut variabel ketiga. Saya sudah sangat dekat dengan ini:
plt.scatter(w, M, c=p, marker='s')di mana w dan M adalah datapoints dan p adalah variabel yang ingin saya beri naungan.
Namun saya ingin melakukannya dalam skala abu-abu daripada warna. Adakah yang bisa membantu?
Jawaban:
Tidak perlu mengatur warna secara manual. Sebagai gantinya, tentukan colormap grayscale ...
import numpy as np
import matplotlib.pyplot as plt
# Generate data...
x = np.random.random(10)
y = np.random.random(10)
# Plot...
plt.scatter(x, y, c=y, s=500)
plt.gray()
plt.show()
Atau, jika Anda lebih suka rentang colormaps yang lebih luas , Anda juga dapat menentukan cmapkwarg scatter. Untuk menggunakan versi yang dibalik dari semua ini, cukup tentukan versi " _r" dari salah satunya. Misalnya gray_rbukannya gray. Ada beberapa colormaps grayscale berbeda pra-dibuat (misalnya gray, gist_yarg, binary, dll).
import matplotlib.pyplot as plt
import numpy as np
# Generate data...
x = np.random.random(10)
y = np.random.random(10)
plt.scatter(x, y, c=y, s=500, cmap='gray')
plt.show()mpl.cmjuga tersedia langsung sebagai plt.cm.
plt.contour()atau plt.contourf()- tapi itu pertanyaan yang berbeda
Dalam matplotlib, warna abu-abu dapat diberikan sebagai string dengan nilai numerik antara 0-1.
Sebagai contohc = '0.1'
Kemudian Anda bisa mengonversi variabel ketiga Anda dalam nilai di dalam rentang ini dan menggunakannya untuk mewarnai poin Anda.
Dalam contoh berikut ini saya menggunakan posisi y titik sebagai nilai yang menentukan warna:
from matplotlib import pyplot as plt
x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [125, 32, 54, 253, 67, 87, 233, 56, 67]
color = [str(item/255.) for item in y]
plt.scatter(x, y, s=500, c=color)
plt.show()
cargumen, dengan nilai alpha di kolom keempat. Jika variabel ketiga adalah Z, dengan shape=(n,1), maka colors = numpy.hstack((numpy.zeros_like(z), numpy.zeros_like(z), numpy.ones_like(z), z/z.max()))memberikan efek yang sangat bagus (tentu saja bisa di-tweak).
plt.colorbar()akan melakukan itu
Terkadang Anda mungkin perlu membuat plot warna dengan tepat berdasarkan case x-value . Misalnya, Anda mungkin memiliki kerangka data dengan 3 jenis variabel dan beberapa titik data. Dan Anda ingin melakukan hal berikut,
Dalam hal ini, Anda mungkin harus menulis ke fungsi pendek untuk memetakan nilai x ke nama warna yang sesuai sebagai daftar dan kemudian meneruskan daftar itu ke plt.scatterperintah.
x=['A','B','B','C','A','B']
y=[15,30,25,18,22,13]
# Function to map the colors as a list from the input list of x variables
def pltcolor(lst):
cols=[]
for l in lst:
if l=='A':
cols.append('red')
elif l=='B':
cols.append('blue')
else:
cols.append('green')
return cols
# Create the colors list using the function above
cols=pltcolor(x)
plt.scatter(x=x,y=y,s=500,c=cols) #Pass on the list created by the function here
plt.grid(True)
plt.show()