如何用matplotlib绘制numpy数组每行对应一条竖线的垂直折线图

你可以直接生成列索引作为y轴坐标，逐行绘制即可实现x/y轴交换的垂直折线效果，示例代码如下：

import numpy as np
import matplotlib.pyplot as plt

# 生成示例矩阵
matrix = np.random.rand(10, 10)
# 生成y轴坐标：长度和矩阵列数一致的连续索引
y_coord = np.arange(matrix.shape[1])

# 遍历每一行绘制垂直折线
for row in matrix:
    plt.plot(row, y_coord)

plt.show()

如果你需要和参考图类似的、各折线之间有固定偏移避免重叠的效果，可以在绘制时给每一行的x值加上对应偏移量，示例如下：

import numpy as np
import matplotlib.pyplot as plt

matrix = np.random.rand(10, 10)
y_coord = np.arange(matrix.shape[1])
offset = 0.3 # 可自行调整偏移大小

for idx, row in enumerate(matrix):
    # 给每行x值加上偏移量
    plt.plot(row + idx * offset, y_coord)

plt.show()

原理解释：直接调用plt.plot(matrix)时，matplotlib默认将矩阵的行索引作为x轴、矩阵元素值作为y轴，按列绘制折线。我们手动逐行绘制时，将行元素值作为第一个参数（x轴）、索引序列作为第二个参数（y轴），就直接实现了坐标轴交换的效果。

内容的提问来源于stack exchange，提问作者B.Quaink