Matplotlib розсіювач; колір як функція третьої змінної

Я хочу зробити розсип (з використанням matplotlib), де точки заштриховані відповідно до третьої змінної. Я дуже близько з цим:

plt.scatter(w, M, c=p, marker='s')

де w і M - точки даних, а p - змінна, яку я хочу відтінити стосовно.
Однак я хочу зробити це в кольорах сірого, а не в кольорі. Хтось може допомогти?

Не потрібно вручну встановлювати кольори. Замість цього вкажіть кольорову карту сірого ...

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

Або, якщо ви віддаєте перевагу більш широкому діапазону кольоровихcmap карт , ви можете також вказати kwarg scatter. Щоб використовувати зворотну версію будь-якого з них, просто вкажіть " _r" версію будь-якої з них. Наприклад, gray_rзамість gray. Є кілька різних відтінки сірих колірні карт попередньо зроблених (наприклад gray, gist_yarg, binaryі т.д.).

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

У matplotlib сірі кольори можуть бути задані у вигляді рядка числового значення між 0-1.
Наприкладc = '0.1'

Тоді ви можете конвертувати третю змінну у значення всередині цього діапазону та використовувати її для розфарбовування балів.
У наступному прикладі я використав y положення точки як значення, яке визначає колір:

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

Іноді вам може знадобитися накреслити колір саме на підставі випадку x-значення . Наприклад, у вас може бути кадр даних з 3 типами змінних та деякими точками даних. І ви хочете зробити наступне,

• Точки графіку, що відповідають фізичній змінній 'A' в RED.
• Точки графіку, що відповідають фізичній змінній 'B' у BLUE.
• Графічні точки, що відповідають фізичній змінній 'C' у ЗЕЛЕНОМ.

У цьому випадку вам, можливо, доведеться записати на коротку функцію, щоб зіставити значення x у відповідні імена кольорів як список, а потім передати цей список plt.scatterкоманді.

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