matplotlib 分散需要太长时间

Question

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

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 点。 但是，这样做需要很长时间，我怎样才能使它更快？ 谢谢

Answer 1

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

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

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