Matplotlib散點圖;顏色作爲第三個變量的函數

Question

我想製作一個散點圖（使用matplotlib），其中的點根據第三個變量着色。我已經非常接近這一點：

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

其中w和M是數據點，p是我想遮蔽的變量。
但是我想用灰度而不是顏色來做。誰能幫忙？

Answer 1

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

Answer 2

沒有必要手動設置的顏色。相反，指定的灰度顏色表...

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

或者，如果你願意一個wider range of colormaps，你還可以指定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()