我要绘制一系列线(目前总共60条线),以绘制到同一图中以显示某个过程的时间演变。当前已绘制线,因此最早的时间步长以100%红色绘制,最新的时间步长以100%蓝色绘制,中间的时间步长根据时间分别为红色和蓝色(红色的数量随时间增加线性减少,而蓝色的数量随时间增加线性增加;简单的颜色渐变)。我想制作一种(最好是垂直的)颜色条,以连续的方式显示出来。我的意思是,我想要一个颜色条,该颜色条的底部是红色,顶部是蓝色,中间是红色和蓝色的混合物,以与线条相同的方式从红色平滑过渡到蓝色我策划我还想将轴放在此颜色栏上,以便可以显示哪种颜色对应于哪个时间步。
我已经阅读了matplotlib.pyplot.colorbar()的文档,但是如果不使用matplotlib先前定义的颜色图之一,就无法弄清楚该怎么做。我的猜测是,我需要定义自己的颜色图,该颜色图从红色过渡到蓝色,然后将其提供给matplotlib.pyplot.colorbar()并获得所需的颜色条相对简单。
这是我用来绘制线条的代码示例:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15import numpy as np
from matplotlib import pyplot as pp
x= ##x-axis values for plotting
##start_time is the time of the earliest timestep I want to plot, type int
##end_time is the time of the latest timestep I want to plot, type int
for j in range(start_time,end_time+1):
##Code to read data in from time step number j
y = ##the data I want to plot
red = 1. - (float(j)-float(start_time))/(float(end_time)-float(start_time))
blue = (float(j)-float(start_time))/(float(end_time)-float(start_time))
pp.plot(bin_center,spectrum,color=(red,0,blue))
pp.show()
编辑:
也许这会使我的意思更清楚。下面是我的图:
每行显示了在不同时间步长下x值和y值之间的关系。红线是较早的时间步,蓝线是较晚的时间步,紫色线在中间,如我上面所定义。使用已经绘制的图形,我如何创建一个颜色栏(如果可能,请在右侧垂直)将线的颜色(连续)映射到每种颜色所代表的时间步长的时间值?
看看这个问题的答案
@TheDude发生的事情太多了,我在回答中不明白这对我来说是不可能知道如何将其转化为我的问题的。
您需要在这里做两件事。
创建一个红色到蓝色的颜色映射,因为它不是matplotlib.cm中的标准映射之一。
创建从时间值到matplotlib.pyplot.plot()可以识别的颜色值的映射。
目前,您基本上是通过red和blue变量来完成这两个操作的,这对plt.plot()很好,但是plt.colorbar()将需要此信息作为Matplotlib可映射对象,例如ScalarMappable 。 如果在开始绘制之前设置了该对象,则还可以使用它在plt.plot()调用中选择适当的颜色。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as mcol
import matplotlib.cm as cm
start_time = 100
end_time = 120
# Generate some dummy data.
tim = range(start_time,end_time+1)
xdat = np.arange(0,90.1)
ydat = [np.sin(0.2*(xdat-t)/np.pi) for t in tim]
# Make a user-defined colormap.
cm1 = mcol.LinearSegmentedColormap.from_list("MyCmapName",["r","b"])
# Make a normalizer that will map the time values from
# [start_time,end_time+1] -> [0,1].
cnorm = mcol.Normalize(vmin=min(tim),vmax=max(tim))
# Turn these into an object that can be used to map time values to colors and
# can be passed to plt.colorbar().
cpick = cm.ScalarMappable(norm=cnorm,cmap=cm1)
cpick.set_array([])
F = plt.figure()
A = F.add_subplot(111)
for y, t in zip(ydat,tim):
A.plot(xdat,y,color=cpick.to_rgba(t))
plt.colorbar(cpick,label="Time (seconds)")
当我运行上面的内容时,Python声称label不是plt.colorbar()的关键字参数。 如果我删除标签部分,尽管一切运行正常。
Matplotlib v1.3.0(2013年5月)中似乎添加了label,因此该特定位不适用于早期版本。