Python matplotlib 用不同颜色绘制数组

Question

我有一个文件，它组合了来自不同来源的数据集，但保留了通用格式。我想使用不同的颜色来绘制这些数据以指示数据集的不同来源。例如，数据文件中的几行如下所示:

# Source measurement1 measurement2 error color
SiteA  543.2 12.3 0.01 blue
SiteB  545.6 12.5 0.02 red
SiteA  545.9 12.9 0.01 blue
SiteC  549.1 13.2 0.01 orange
SiteB  550.4 13.3 0.02 red
...

目前我做了一个 for 循环并绘制每个点:

for point in data:
   plt.errorbar(measurement1,measurement2,yerr=error, marker='.', ecolor='k', fmt=color, linestyle='.')

这会单独绘制每个点，但对于大型数据数组可能需要很长时间。

谁能建议一种更快的方法吗？

Answer 1

如果您没有太多颜色，您应该能够通过按颜色组进行绘图来加快速度，即为每种颜色调用一次pyplot.errorbar()。使用列表理解将数据分组为颜色组，并为测量1、测量2等提供列表或数组而不是标量。