最近在可视化一些数据,需要将坐标转化成直线,看了很多教程发现很多是错的,现在记录一下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()
运行结果: