Matplotlib scatterplot; kolor jako funkcja trzeciej zmiennej

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?

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()

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()

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ć,

• Wykreśl punkty odpowiadające zmiennej fizycznej „A” zaznaczonej na czerwono.
• Wykreśl punkty odpowiadające zmiennej fizycznej „B” na NIEBIESKO.
• Wykreśl punkty odpowiadające zmiennej fizycznej „C” na ZIELONO.

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()