Matplotlib scatterplot; màu là hàm của biến số thứ ba

Question

Tôi muốn tạo một phân tán (sử dụng matplotlib) trong đó các điểm được tô bóng theo biến thứ ba. Tôi đã rất gần với điều này:

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

trong đó w và M là datapoints và p là biến tôi muốn tô bóng liên quan.
Tuy nhiên tôi muốn làm điều đó trong thang độ xám hơn là màu. Có ai giúp được không?

Answer 1

Trong màu xám matplotlib có thể được xem như một chuỗi có giá trị bằng số từ 0-1.
Ví dụ: c = '0.1'

Sau đó, bạn có thể chuyển đổi biến thứ ba của mình thành giá trị bên trong phạm vi này và sử dụng biến đó để tô màu điểm của bạn.
Trong ví dụ sau tôi đã sử dụng vị trí y của điểm như giá trị mà xác định màu:

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

Answer 2

Không cần phải tự thiết lập các màu sắc. Thay vào đó, chỉ định một bản đồ màu xám ...

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

Hoặc, nếu bạn muốn một wider range of colormaps, bạn cũng có thể chỉ định cmap kwarg để scatter. Để sử dụng phiên bản đảo ngược của bất kỳ phiên bản nào trong số này, chỉ cần chỉ định phiên bản "_r" của bất kỳ phiên bản nào trong số này. Ví dụ. gray_r thay vì gray. Có một số bản đồ màu xám khác nhau được tạo sẵn (ví dụ: gray, gist_yarg, binary, v.v ...).

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