python - 如何对 pyplot.plot 使用可变数字参数？

Question

这是我的数据和我想要生成的图表:

from numpy import array
from os.path import expanduser
import matplotlib.pyplot as plt
sns.set(style="darkgrid")

data = [array([8, 4, 9, 6, 2]), array([5, 4, 4, 1, 2]), array([4, 3, 1, 5, 6]), array([8, 3, 5, 6, 4])]

for d in data:
    plt.plot(d)

# plt.savefig(expanduser("~/Documents/rural_juror.pdf"))

因为我想在同一个 seaborn facetgrid 中绘制其中的几个，所以我不能使用循环，但必须在一行中进行，如下所示:

plt.plot(*data)
# produces same result as 
# plt.plot(data[0], data[1], data[2], data[3])

然而，这不起作用，但会生成如下图:

我猜这与 plot 的参数类型有关 data, looks, data, looks... 但我该如何解决这个问题？

Answer 1

是的，如果您想使用单行代码，则需要提供 x 和 y 值。你可以这样做:

import numpy as np
x = np.arange(5)
data = [x, array([8, 4, 9, 6, 2]),
        x, array([5, 4, 4, 1, 2]),
        x, array([4, 3, 1, 5, 6]),
        x, array([8, 3, 5, 6, 4])]
plt.plot(*data)