So erstellen Sie ein Streudiagramm aus einem Pandas DataFrame

Es gibt zwei Möglichkeiten, ein Streudiagramm mit Daten aus einem Pandas DataFrame zu erstellen:

1. Verwenden Sie pandas.DataFrame.plot.scatter

Eine Möglichkeit, ein Streudiagramm zu erstellen, besteht darin, die integrierte pandas Funktion plot.scatter() zu verwenden:

import pandas as pd

df.plot.scatter(x = 'x_column_name', y = 'y_columnn_name')

2. Verwenden Sie matplotlib.pyplot.scatter

Eine andere Möglichkeit, ein Streudiagramm zu erstellen, ist die Verwendung der Funktion Matplotlib pyplot.scatter(). Es ist anzumerken, dass die obige pandas Funktion plot.scatter() unter der Haube ebenfalls die matplotlib Bibliothek zum Diagramm verwendent.

import matplotlib.pyplot as plt

plt.scatter(df.x, df.y)

Dieses Tutorial enthält ein Beispiel für die Verwendung jeder dieser Methoden.

Beispiel 1: Verwenden Sie Pandas

Der folgende Code zeigt, wie Sie mit der Funktion plot.scatter() ein einfaches Streudiagramm erstellen:

import pandas as pd

# Dataframe erstellen
df = pd.DataFrame({'x': [1, 3, 3, 4, 5, 7, 9, 12, 13, 15],
                   'y': [5, 7, 9, 7, 6, 12, 14, 18, 15, 22]})

# Streudiagramm erstellen
df.plot.scatter(x='x', y='y')

Beachten Sie, dass Sie die Argumente s und c verwenden können, um die Größe und Farbe der Punkte zu ändern:

df.plot.scatter(x='x', y='y', s=60, c='green')

Beispiel 2: Verwenden Sie Matplotlib

Der folgende Code zeigt, wie Sie mit der Funktion pyplot.scatter() ein Streudiagramm erstellen:

import pandas as pd
import matplotlib.pyplot as plt

# Dataframe erstellen
df = pd.DataFrame({'x': [1, 3, 3, 4, 5, 7, 9, 12, 13, 15],
                   'y': [5, 7, 9, 7, 6, 12, 14, 18, 15, 22]})

# Streudiagramm erstellen
plt.scatter(df.x, df.y)

Beachten Sie, dass Sie die Argumente s und c verwenden können, um die Größe und Farbe der Punkte zu ändern:

plt.scatter(df.x, df.y, s=60, c='purple')

Weitere Python-Tutorials finden Sie hier.