使用 matplotlib 绘制两个不同大小的数组

Question

我想使用 matplotlib 来绘制我的数据。 我想在同一张图上绘制 4 个不同的 y 值，以便我可以比较这 4 个。目前，我的“x”数据是一个大小为 10 的数组：

x: [1,2,3,4,5,6,7,8]

我的 y 数据有 3 个值，是：

y: [[0.6, 0.3, 0.4], [0.2, 0.5, 0.4], [0.6, 0.3, 0.4], .... etc.]

每个嵌套数组对应于该 x 值。 因此，对于 x=1，不同的 y 值为 0.6、0.3、0.4。

我怎样才能把这些都画在一张图上？

谢谢。

Answer 1

您可以在绘图前将列表 (y) 列表转换为更合适的格式，并使用 for 循环在同一图形中绘图。

试试这个例子（是基本的）：

def plotting(y):
    new_y = [ [sublist[i] for sublist in y] for i in range(len(y[0])) ]

    fig, axes = plt.subplots(1,1)
    for sublist in new_y:
        x = np.arange(0, len(sublist))
        axes.plot(x, sublist)

    return fig, axes

请注意，该方法的第一行创建了一个新的 y 列表。

# use
fig, ax = plotting(y)

Answer 2

使用zip函数转置y ，提取大小等于x数组。 例如，

y1,y2,y3=list(zip(*y))

然后分别绘制 y1、y2 和 y3。

Answer 3

你可以只做一个 for 循环：

for col in np.array(y).T:
    plt.plot(x, col)

在示例数据上运行：

x = [1,2,3,4,5,6,7,8]
y = [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11],
     [12, 13, 14], [15, 16, 17], [18, 19, 20], [21, 22, 23]
    ]

输出：