[Python] Wie zeichnet man mit Matplotlib ein Streudiagramm?
Verwenden Sie Streuung, um ein Streudiagramm zu zeichnen. Einige Beispiele sind unten gezeigt.
Einfaches Streudiagramm
Unten sehen Sie ein Beispiel für das einfachste Streudiagramm.
import numpy as np
import matplotlib.pyplot as plt
# generate data
x = np.random.rand(100)
y = np.random.rand(100)
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.scatter(x,y)
ax.set_title('first scatter plot')
ax.set_xlabel('x')
ax.set_ylabel('y')
fig.show()
Ändern Sie die Farbe für jede Daten
Sie können die Farbe ändern, indem Sie im Parameter c = 'red' angeben.
import numpy as np
import matplotlib.pyplot as plt
# generate data
x1 = np.random.rand(100)*0.5
y1 = np.random.rand(100)
x2 = np.random.rand(100)*0.5 + 0.5
y2 = np.random.rand(100)
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.scatter(x1,y1, c='red')
ax.scatter(x2,y2, c='blue')
ax.set_title('second scatter plot')
ax.set_xlabel('x')
ax.set_ylabel('y')
fig.show()
Das gleiche Ergebnis wird erhalten, auch wenn es in RGB angegeben ist. Geben Sie zu diesem Zeitpunkt einen Wert von 0,0 bis 1,0 an.
ax.scatter(x1,y1, c=(1.0,0,0))
ax.scatter(x2,y2, c=(0, 0, 1.0))
Legende und Gitterlinie hinzufügen
Die Legende verwendet Legende. Sie können die Anzeigeposition mit einem Argument ändern. Wenn Sie eine Gitterlinie zeichnen möchten, verwenden Sie das Gitter (True).
| Position |
|---|
| upper right |
| upper left |
| lower left |
| lower right |
| right |
| center left |
| center right |
| lower center |
| upper center |
| center |
import numpy as np
import matplotlib.pyplot as plt
# generate data
x1 = np.random.rand(100)*0.5
y1 = np.random.rand(100)*0.5
x2 = np.random.rand(100)*0.5 + 0.5
y2 = np.random.rand(100)*0.5
x3 = np.random.rand(100)*0.5
y3 = np.random.rand(100)*0.5 + 0.5
x4 = np.random.rand(100)*0.5 + 0.5
y4 = np.random.rand(100)*0.5 + 0.5
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.scatter(x1,y1, c='red', label='group1')
ax.scatter(x2,y2, c='blue', label='group2')
ax.scatter(x3,y3, c='green', label='group3')
ax.scatter(x4,y4, c='yellow', label='group4')
ax.set_title('third scatter plot')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.grid(True)
ax.legend(loc='upper left')
fig.show()
Ändern Sie die Markierung
Marker werden als marker = 'o' angegeben. Als Beispiel wurden vier typische Marker verwendet. Es gibt viele andere. Siehe hier.
import numpy as np
import matplotlib.pyplot as plt
# generate data
x1 = np.random.rand(100)*0.5
y1 = np.random.rand(100)*0.5
x2 = np.random.rand(100)*0.5 + 0.5
y2 = np.random.rand(100)*0.5
x3 = np.random.rand(100)*0.5
y3 = np.random.rand(100)*0.5 + 0.5
x4 = np.random.rand(100)*0.5 + 0.5
y4 = np.random.rand(100)*0.5 + 0.5
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.scatter(x1,y1, c='red', marker='.', label='group1')
ax.scatter(x2,y2, c='blue',marker='o', label='group2')
ax.scatter(x3,y3, c='green',marker='^', label='group3')
ax.scatter(x4,y4, c='yellow',marker='s', label='group4')
ax.set_title('fourth scatter plot')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.grid(True)
ax.legend(loc='upper left')
fig.show()
Ändern Sie die Größe des Markers
Die Größe des Markers ist ein Parameter wie s = 20. Die Standardgröße ist 20.
import numpy as np
import matplotlib.pyplot as plt
# generate data
x1 = np.random.rand(100)*0.5
y1 = np.random.rand(100)*0.5
x2 = np.random.rand(100)*0.5 + 0.5
y2 = np.random.rand(100)*0.5
x3 = np.random.rand(100)*0.5
y3 = np.random.rand(100)*0.5 + 0.5
x4 = np.random.rand(100)*0.5 + 0.5
y4 = np.random.rand(100)*0.5 + 0.5
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
ax.scatter(x1,y1, c='red', s=20, marker='o', label='group1')
ax.scatter(x2,y2, c='blue',s=40, marker='o', label='group2')
ax.scatter(x3,y3, c='green',s=80, marker='o', label='group3')
ax.scatter(x4,y4, c='yellow',s=120, marker='o', label='group4')
ax.set_title('fifth scatter plot')
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.grid(True)
ax.legend(loc='upper left')
fig.show()
Recommended Posts