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~
