Streudiagramme in Pandas / Pyplot: So zeichnen Sie nach Kategorie

Ich versuche, mit einem Pandas DataFrame-Objekt ein einfaches Streudiagramm in Pyplot zu erstellen, möchte jedoch eine effiziente Methode zum Zeichnen von zwei Variablen, deren Symbole jedoch durch eine dritte Spalte (Schlüssel) vorgegeben werden. Ich habe verschiedene Methoden mit df.groupby ausprobiert, aber nicht erfolgreich. Ein Beispiel für ein df-Skript finden Sie unten. Dies färbt die Markierungen gemäß 'key1', aber ich würde gerne eine Legende mit 'key1' Kategorien sehen. Bin ich nah Vielen Dank.

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
df = pd.DataFrame(np.random.normal(10,1,30).reshape(10,3), index = pd.date_range('2010-01-01', freq = 'M', periods = 10), columns = ('one', 'two', 'three'))
df['key1'] = (4,4,4,6,6,6,8,8,8,8)
fig1 = plt.figure(1)
ax1 = fig1.add_subplot(111)
ax1.scatter(df['one'], df['two'], marker = 'o', c = df['key1'], alpha = 0.8)
plt.show()

Antworten auf die Frage(6)

Ihre Antwort auf die Frage