python - 逐行绘制颜色图

Question

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

在我的两列文件中，时间将由行号描述。要么写为文件中的第 3 列，要么使用 pyplot 的固有功能自行获取行号。

谁能帮我做这件事？

非常感谢。

尼古拉斯

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