Python绘制颜色

Question

通过研究，我找到了使用迭代器和循环绘制n数据集的答案。

有没有不使用循环的方法？

我有一个数组odata，其中包含n列。第一列是横坐标。我想把剩下的画出来。

我有这样一行代码：

plt.plot(odata[:,0],odata[:,1:],'-')

这会进行打印，但我想添加自己的颜色和标签。怎样

Answer 1

您可以访问使用plt.plot的返回值创建的Line2D对象，并根据该列表直接设置图例：

labels = ['a', 'b', 'c']
lines = plt.plot(odata[:, 0], odata[:, 1:], '-')
plt.legend(lines, labels)

标签和行的数量不一定要匹配。如果行数较少，则某些标签将不使用。如果标签较少，则某些行将未标记。以下是文档中的图例指南。

为了无缝地改变颜色的顺序，您需要通过matplotlib.rc在全局配置中设置循环器，就像在文档中的这个例子一样，或者使用面向对象的API来绘制。然后，您可以在单个坐标轴上使用set_prop_cycle，而无需摆弄全局设置。

这里有三种方法来设置颜色循环器，以增加个人偏好。请注意，我只是在这里展示如何设置颜色，但您也可以控制线条样式的顺序，可能还有其他属性：

>

设置全局配置：

import matplotlib as mpl
from matplotlib import cycler
from matplotlib import pyplot as plt

labels = ['a', 'b', 'c']
colors = ['r', 'g', 'b']

mpl.rc('axes', prop_cycle=cycler('color', ['r', 'g', 'b', 'y']))

lines = plt.plot(odata[:, 0], odata[:, 1:], '-')
plt.legend(lines, labels)

设置全局配置，但使用rc_context上下文管理器使更改有效地本地到您的情节：

import matplotlib as mpl
from matplotlib import cycler,rc_context
from matplotlib import pyplot as plt

labels = ['a', 'b', 'c']
colors = ['r', 'g', 'b']

with rc_context(rc={'axes.prop_cycle': cycler('color', ['r', 'g', 'b', 'y'])}):
    lines = plt.plot(odata[:, 0], odata[:, 1:], '-')
    plt.legend(lines, labels)

首先使用面向对象的API设置绘图，并仅对实际要修改的轴应用更改：

from matplotlib import pyplot as plt

labels = ['a', 'b', 'c']
colors = ['r', 'g', 'b']

fig, ax = plt.subplots()
ax.set_prop_cycle('color', colors)
ax.plot(odata[:, 0], odata[:, 1:], '-')
ax.legend(labels)

我建议将面向对象的API作为一般规则，特别是在独立脚本中，因为它提供了更大的灵活性，并且在确切知道将对哪些对象进行操作方面也很清晰。