时间戳

首页 > 解决方案 > matplotlib scatter 耗时太长

python - matplotlib scatter 耗时太长

问题描述

考虑下面的代码，它产生各种颜色的点的随机序列。

import matplotlib.pyplot as plt
import random

n = 100

x = [0]*n
y = [0]*n
col=['red']*n


for i in range(1,n): 
    val = random.randint(1,2)
    if val == 1: 
        col[i] = 'red'
    if val == 2: 
        col[i] = 'yellow'

ax=plt.figure().add_subplot()
for i in range(n):
    ax.scatter(i,0, color=col[i])
    #print(x[i],y[i],col[i])
plt.show()

现在它做 100 点，但我希望它做 10000 点。但是这样做需要很长时间，我怎样才能让它更快？谢谢

标签： pythonmatplotlib

解决方案

n避免一次绘制一个散点的不必要的 for 循环。相反，一次绘制整个范围并传递整个col颜色列表

fig = plt.figure()
ax = fig.add_subplot(111)

ax.scatter(range(n), [0]*n, color=col)
plt.show()

推荐阅读