如何让matplotlib使用facecolors='none‘为不同的类绘制不同的边框颜色？

import matplotlib.pyplot as plt
import numpy as np

# 创建数据集
x = np.random.rand(10)
y = np.random.rand(10)
labels = np.random.randint(0, 3, 10)  # 生成随机的类别标签，范围为0到2

# 绘制散点图
plt.scatter(x, y, c=labels, cmap='viridis', edgecolors='black')

# 设置边框颜色
for i, j in enumerate(labels):
    plt.gca().get_children()[i].set_edgecolor(plt.cm.viridis(j))

plt.show()