[PYTHON] Hinweise zur Farbfärbung im Matplotlib-Streudiagramm
Was ist dieser Artikel?
Streuung wird verwendet, wenn ein Streudiagramm mit matplotlib gezeichnet wird. Wenn jedoch jedem Punkt ein Wert zugewiesen wird, gibt es Zeiten, in denen Sie entsprechend diesem Wert färben möchten. Ich vergesse immer, wie man zeichnet, also ist dies ein Memo dafür.
Quellcode
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 steht für einen Punkt in der zweiten Dimension und y für den jedem Punkt zugewiesenen Wert. Dieses Mal habe ich jedem Punkt einen Wert von -1 bis 1 zugewiesen. Durch Angabe von vmin und vmax in den an plt.scatter übergebenen Argumenten können Sie die minimalen und maximalen Werte von colormap angeben. Sie können den Farbbalken der Farbkarte auch zeichnen, indem Sie den Rückgabewert von plt.scatter an plt.colorbar übergeben.
Ausführungsergebnis
Recommended Posts