matplotlib - scatterplot, choisissez des couleurs pour les tuples

Question

Je suis nouveau sur Python et j'essaye de créer un nuage de points montrant plusieurs valeurs y pour chaque xtick. J'ai trouvé cette question qui m'a aidé à trouver une sorte de code: Python Scatter Plot with Multiple Y values for each X.

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() 

Le problème est que, par exemple, ce code imprime alternativement les points du 1er tuple en rouge et bleu. Cependant, j'aime imprimer tous les points du 1er tuple (par exemple (1,1,2,3,9)) en bleu, tous les points du tuple suivant (par exemple (1,1,2,4)) en rouge etc. Y a-t-il un moyen de spécifier la couleur de cette manière? Je n'ai pas encore rencontré une question similaire.

J'apprécie vraiment n'importe quel conseil ou aide!

Answer 1

Utilisez itertools.cycle pour faire défiler vos couleurs, et les zip avec vos tuples dans la boucle:

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