在matplotlib中绘制具有不同颜色的样本

Question

我尝试使用以下代码在2D平面上绘制一组样本。

import numpy as np
import matplotlib
import matplotlib.pyplot as plt

# some code generating x and y

matplotlib.rcParams['axes.unicode_minus'] = False
fig, ax = plt.subplots()
ax.plot(x, y, 'o')
ax.set_title('Using hypen instead of unicode minus')
plt.show()  

我想提供一个值列表（每个值的范围从0到k-1），以便每个样本都显示相应的颜色。我怎么需要修改上面的代码？