在matplotlib中绘制大量的点和边

时间:2011-11-09 16:32:12

标签: python numpy matplotlib

我有一些点(大约3000)和边缘(大约6000)这种格式:

points = numpy.array([1,2],[4,5],[2,7],[3,9],[9,2])
edges = numpy.array([0,1],[3,4],[3,2],[2,4])

其中边是索引到点,因此每条边的起点和终点坐标由下式给出:

points[edges]

我正在寻找一种更快/更好的方式来绘制它们。目前我有:

from matplotlib import pyplot as plt
x = points[:,0].flatten()
y = points[:,1].flatten()
plt.plot(x[edges.T], y[edges.T], 'y-') # Edges
plt.plot(x, y, 'ro') # Points
plt.savefig('figure.png')

我读到了关于lineCollections的信息,但我不确定如何使用它们。有没有办法更直接地使用我的数据?这里的瓶颈是什么?

一些更现实的测试数据,绘图时间约为132秒:

points = numpy.random.randint(0, 100, (3000, 2))
edges = numpy.random.randint(0, 3000, (6000, 2))

2 个答案:

答案 0 :(得分:3)

好吧,我发现以下内容要快得多:

from matplotlib import pyplot as plt
from matplotlib.collections import LineCollection
lc = LineCollection(points[edges])
fig = plt.figure()
plt.gca().add_collection(lc)
plt.xlim(points[:,0].min(), points[:,0].max())
plt.ylim(points[:,1].min(), points[:,1].max())
plt.plot(points[:,0], points[:,1], 'ro')
fig.savefig('full_figure.png')

是否仍然可以更快地完成它?

答案 1 :(得分:2)

您也可以在单个绘图调用中执行此操作,这比两个快得多(尽管可能与添加LineCollection基本相同)。

import numpy
import matplotlib.pyplot as plt

points = numpy.array([[1,2],[4,5],[2,7],[3,9],[9,2]])
edges = numpy.array([[0,1],[3,4],[3,2],[2,4]])

x = points[:,0].flatten()
y = points[:,1].flatten()

plt.plot(x[edges.T], y[edges.T], linestyle='-', color='y',
        markerfacecolor='red', marker='o') 

plt.show()