Chcę utworzyć wykres rozrzutu (za pomocą matplotlib), w którym punkty są cieniowane zgodnie z trzecią zmienną. Jestem z tym bardzo blisko:
plt.scatter(w, M, c=p, marker='s')gdzie wi M są punktami danych, a p jest zmienną, w odniesieniu do której chcę zacienić.
Jednak chcę to zrobić w odcieniach szarości, a nie w kolorze. Czy ktoś może pomóc?
Odpowiedzi:
Nie ma potrzeby ręcznego ustawiania kolorów. Zamiast tego określ mapę kolorów w skali szarości ...
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()
Lub, jeśli wolisz szerszy zakrescmap map kolorów , możesz również określić kwarg na scatter. Aby użyć odwróconej wersji któregokolwiek z nich, po prostu określ " _r" wersję dowolnego z nich. Np. gray_rZamiast gray. Istnieje kilka różnych colormaps skali szarości gotowych (na przykład gray, gist_yarg, binaryitp).
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.cmjest również dostępny bezpośrednio jako plt.cm.
plt.contour()lub plt.contourf()- ale to inne pytanie
W matplotlib szare kolory można podać jako ciąg o wartości liczbowej z zakresu od 0 do 1.
Na przykładc = '0.1'
Następnie możesz przekonwertować trzecią zmienną na wartość z tego zakresu i użyć jej do pokolorowania punktów.
W poniższym przykładzie użyłem pozycji y punktu jako wartości określającej kolor:
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()
cargument podasz tablicę w kształcie (n, 4) , z wartościami alfa w czwartej kolumnie. Jeśli trzecia zmienna to Z, shape=(n,1)to colors = numpy.hstack((numpy.zeros_like(z), numpy.zeros_like(z), numpy.ones_like(z), z/z.max()))daje bardzo fajny efekt (oczywiście można ją podkręcić).
plt.colorbar()to zrobi
Czasami może zajść potrzeba precyzyjnego wykreślenia koloru na podstawie przypadku wartości x . Na przykład możesz mieć ramkę danych z 3 typami zmiennych i niektórymi punktami danych. I chcesz śledzić,
W takim przypadku może być konieczne napisanie do funkcji short, aby zmapować wartości x do odpowiednich nazw kolorów jako listę, a następnie przekazać tę listę do plt.scatterpolecenia.
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()