Pyplot colormap逐行

Question

我开始使用非常漂亮的pyplot绘制python.我的目标是随着时间的推移展示两个系列数据的演变.我想要一个散点图(data1,data2),其中时间分量显示为颜色渐变,而不是随意绘制数据时间函数.

在我的两个列文件中,时间将由行号描述.要么使用pyplot的内在功能写入文件中的第3列,要么自行获取行号.

谁能帮助我做到这一点？

非常感谢.

萨科

Answer 1

使用时,matplotlib.pyplot.scatter您可以通过关键字参数传递第三个数组c.此数组可以选择散射点所需的颜色.然后,您还可以从中选择适当的颜色映射,matplotlib.cm并使用cmap关键字参数进行分配.

这个玩具示例创建了两个数据集data1和data2.然后它还创建一个数组colors,一个在0和1之间等间隔的连续值数组,其长度与data1和相同data2.它不需要知道"行号",它只需要知道数据点的总数,然后同样地分隔颜色.

我还添加了一个colorbar.您可以通过删除该plt.colorbar()行来删除它.

import matplotlib.pyplot as plt
from matplotlib import cm
import numpy as np


N = 500
data1 = np.random.randn(N)
data2 = np.random.randn(N)
colors = np.linspace(0,1,N)

plt.scatter(data1, data2, c=colors, cmap=cm.Blues)

plt.colorbar()

plt.show()