如何使用matplotlib绘制不同颜色的动画点？

Question

我有一个生成动画点的函数，这是导致问题的部分：

<name>.png

所以，dot是我的对象类的名称et dots是包含N个对象的列表。每个点都用红色绘制。

我想要做的是绘制，例如，红色的N-1点和蓝色的一个点，是否可以使用命令ax.plot？

感谢您的帮助

Answer 1

是的，有可能。您需要将这些点分成两个集合;有很多方法可以做到这一点;在这里，我选择从列表中提取一个点。那么你必须在同一个画布上分别绘制每个集合。

import random
import matplotlib.pyplot as plt


class Dot(object):
    def __init__(self, x, y):
        self.x = x
        self.y = y

def get_random_dot(dots):
    random.shuffle(dots)
    return dots.pop()

num_dots = 10
dots = [Dot(random.random(), random.random()) for _ in range(num_dots)] 

fig = plt.figure()  
ax = plt.axes() 

selected_dot = get_random_dot(dots)
d, = ax.plot([dot.x for dot in dots],[dot.y for dot in dots], 'r.')
f, = ax.plot(selected_dot.x, selected_dot.y, color='blue', marker='o', linewidth=3)

plt.show()