[PYTHON] Remarques sur la coloration par valeur dans le diagramme de dispersion matplotlib
Quel est cet article?
Le nuage de points est utilisé lors du dessin d'un diagramme de dispersion avec matplotlib, mais lorsqu'une valeur est attribuée à chaque point, il y a des moments où vous souhaitez colorer en fonction de cette valeur. J'oublie toujours comment dessiner, c'est donc un mémo pour ça.
Code source
import numpy as np
import matplotlib.cm as cm
import matplotlib.pyplot as plt
if __name__ == '__main__':
N = 100
X = np.random.rand(N, 2)
y = np.random.rand(N) * 2 - 1
sc = plt.scatter(X[:, 0], X[:, 1], vmin=-1, vmax=1, c=y, cmap=cm.seismic)
plt.colorbar(sc)
plt.show()
X représente un point sur la deuxième dimension et y représente la valeur attribuée à chaque point. Cette fois, j'ai attribué une valeur de -1 à 1 pour chaque point. En spécifiant vmin et vmax dans les arguments passés à plt.scatter, vous pouvez spécifier les valeurs minimum et maximum de colormap. Vous pouvez également dessiner la barre de couleurs de la palette de couleurs en passant la valeur de retour de plt.scatter à plt.colorbar.
Résultat d'exécution
Recommended Posts