对应于值的散点图颜色

Question

在所有的colormap帖子中我都没有找到这个答案,或者可能听不懂.

我想用颜色制作散点图.

我有一个清单B:

[1.29,
 1.27,
 1.46,
 0.91,
 0.56,
 0.99,
 1.00,
 0.37,
 1.24,
 1.23]

我将使用一个愚蠢的例子,如果你这样做:

import matplotlib.pyplot as plt
from matplotlib import cm
from math import sin

x=range(10)
y=[sin(i) for i in x]
colors=np.linspace(0,1,10)

plt.scatter(x,y,c=colors,cmap=cm.jet)

你得到不同颜色的点,很好.

但是!我不想只得到漂亮的彩色点!我希望根据B中值的"强度"对点进行着色.

这是我的愚蠢尝试:

import matplotlib.pyplot as plt
from matplotlib import cm
from math import sin

x=range(10)
y=[sin(i) for i in x]
#colors=np.linspace(0,1,10)
B=[1.29,1.27,1.46,0.91,0.56,0.99,1.00,0.37,1.24,1.23]

plt.scatter(x,y,c=B,cmap=cm.jet)

你可以根据B中值的强度获得颜色,非常好:

但是!! 我想改变颜色的"比例",使深蓝色为0,深红色为2.在这种情况下,第三个点(与之相关B[2]=1.46)应该是橙色而不是深红色!我该怎么做？

Answer 1

您需要使用vmin和vmax参数手动设置最小和最大颜色.对于你的情况,这是,

import matplotlib.pyplot as plt
from matplotlib import cm
from math import sin

x=range(10)
y=[sin(i) for i in x]
#colors=np.linspace(0,1,10)
B=[1.29,1.27,1.46,0.91,0.56,0.99,1.00,0.37,1.24,1.23]

cs = plt.scatter(x,y,c=B,cmap=cm.jet,vmin=0.,vmax=2.)
plt.colorbar(cs)
plt.show()