Matplotlib 折线图 二维数据绘制 (2)

Matplotlib 折线图 二维数据绘制

需求

在使用 python 数据集时, 我们通常需要输出数据集的图像来显示数据集结果。而大部分数据集都是多维度的, 因此我们需要输出多条曲线到同一张图上, 下面我们来实现这个需求。

---

代码

这里我们主要使用了两个库 numpy 以及 matplotlib, 并且在绘制时使用了多组颜色。

注意: plt.legend() 与 plt.show() 需要放在循环外面。

import numpy as np
import matplotlib.pyplot as plt

if __name__ == '__main__':
    datas = np.ones((100,2000))       # 生成 1000 * 2000 的二维数组
    for i in range(datas.shape[0]):
        for j in range(datas.shape[1]):
            datas[i][j] = datas[i][j] * i*j

    y = np.arange(0,datas.shape[1],dtype=int)
    # 画图
    # 颜色
    colors = ['pink','plum','red','seagreen','silver','snow','teal','orange','yellow','blue','black']
    for i in range(datas.shape[0]):
        plt.plot(datas[i], y, c=colors[i % 8], lw=0.5)
        plt.xlabel('x', fontsize=8)
        plt.ylabel('y', fontsize=8)
        plt.title('curve')
    plt.legend(loc='best', fontsize=10)
    plt.show()

---

结果

---

over~