最近在可视化一些数据,需要将坐标转化成直线,看了很多教程发现很多是错的,现在记录一下matplotlib库的pyplot函数的简单用法。

情况1:已知两点坐标

已知:两点坐标 p1,p2 想把他们连接起来,怎么办?

import matplotlib.pyplot as plt
p1 = [5, 23]
p2 = [16, 50]
# 第一个参数是x的取值范围 第二个参数是y的取值范围
plt.plot([p1[0], p2[0]], [p1[1], p2[1]], color='r')
# 画点
plt.scatter([p1[0], p2[0]], [p1[1], p2[1]], color='b')
plt.show()

运行结果:

 

 情况2:有若干个点组成的数组,怎么画线? 

加个遍历即可

import matplotlib.pyplot as plt
p1 = [[560, 883], [589, 428]]
p2 = [[586, 506], (618, 6)]

for i in range(len(p1)):
    plt.plot([p1[i][0], p2[i][0]], [p1[i][1], p2[i][1]], color='r')
    plt.scatter([p1[i][0], p2[i][0]], [p1[i][1], p2[i][1]], color='b')
plt.show()

运行结果: