散点图，为 Y 元组选择颜色

Question

我是Python的新手，我正在尝试创建一个散点图，为每个xtick显示几个y值。我发现了这个问题，它帮助我想出了某种代码：Python 散点图，每个 X 有多个 Y 值。

x = [1,2,3,4]
y = [(1,1,2,3,9),(1,1,2,4), (0.5,4), (0.1,2)]
for xe, ye in zip(x, y):
plt.scatter([xe] * len(ye), ye, color=['blue', 'red'])
plt.xticks([1, 2, 3, 4])
plt.axes().set_xticklabels(['5', '10', '15', '25'])
plt.ylim(0, 1)
plt.show()

问题是，例如，此代码交替以红色和蓝色打印第一个元组的点。但是，我喜欢用蓝色打印第一个元组的所有点(例如(1,1,2,3,9((，用红色打印下一个元组的所有点(例如(1,1,2,4(等。有没有办法以这种方式指定颜色？我还没有遇到类似的问题。

我非常感谢任何提示或帮助！

Answer 1

使用itertools.cycle循环颜色，并将它们与循环中的元组一起压缩：

for (xe, ye), color in zip(zip(x, y), itertools.cycle(['blue', 'red'])):
pyplot.scatter([xe] * len(ye), ye, color=color)