Python pyplot scatter is not using colors - pandas

Question

I am trying to plot a scatter chart with pandas and matplotlib.pylot. The dots in the graph are only using one color, while the legend is showing there are three different colors for three different groups of data.
Below is my code and a copy of screen shot. You can see that only all dots are in green color. Could anyone point me why? What did I do wrong?
Thanks a lot in advance.
import pandas as pd
import matplotlib.pyplot as plt
data = {
'x':[1,2,3,4,1,3,7,5],
'y':[10, 20, 30, 40, 20, 30, 40, 80],
'label':['A', 'A','B','B','A','C','C','A']
}
df = pd.DataFrame(data)
plt.figure(figsize=(34,8))
fig,ax = plt.subplots()
#sns.scatterplot(data=df, hue='label', x='x', y='y')
for k, d in df.groupby('label'):
ax.scatter(df['x'], df['y'], label=k)
plt.legend()
plt.show()

Answer 1

You need to add colors mapping. Slight modifications to your code after adding colors dictionary:
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.cm as cm
data = {
'x':[1,2,3,4,1,3,7,5],
'y':[10, 20, 30, 40, 20, 30, 40, 80],
'label':['A', 'A','B','B','A','C','C','A']
}
df = pd.DataFrame(data)
#plt.figure(figsize=(34,8))
fig,ax = plt.subplots()
df1 = df.groupby('label')
colors = iter(cm.rainbow(np.linspace(0, 1, len(df1.groups))))
for k, d in df1:
ax.scatter(d['x'], d['y'], label=k, color=next(colors))
plt.legend()
plt.show()
outputs the scatter plot as:
Is this your desired output?