Python matplotlib Linecollection 函数用法
今天看到了这个函数的用法觉得很有意思,因为通常,如果我们想要在 matplotlib
中绘制多条线条,通常我们的做法是使用 for
循环绘制。然而当想要绘制的线条逐渐多起来的时候,使用 for
循环会极大拖慢程序运行的速度,此时我们就可以使用 Linecollection()
函数来解决这一问题。
import numpy as np
import matplotlib.pyplot as plt
from matplotlib.collections import LineCollection
lines = np.array([[(0, 1), (1, 1)], [(2, 3), (3, 3)], [(1, 2), (1, 3)]])
colors = ['r', 'g', 'b']
lc = LineCollection(lines, colors=['r', 'g', 'b'])
fig = plt.figure()
ax = fig.add_subplot(111)
ax.add_collection(lc)
ax.autoscale()
plt.show()
最终得到的图像如下图所示:
如果大家觉得有用,就点个赞让更多的人看到吧~