为每个唯一ID创建一个图

Question

Given the dataframe df, I could use some help to create two different scatter plots one for the x,y cordinates, the c value is used for the color map with the id "aa" and one with the x,y cordinates, the c value is used for the color map with the id "bb". With the actual data there are over 1000 unique id's.

import numpy as np
import matplotlib.pyplot as plt
import pyodbc
import pandas as pd


#need to add the 
data = {'x':[2,4,6, 8,10, 12], 'y':[2,4,6, 8,10, 12], 'c': [.2,.5,.5,.7,.8,.9], 'id':['aa','aa','aa','bb','bb','bb']}

df = pd.DataFrame(data)

print (df)

for d in df.groupby(df['id']):

    plt.scatter(d[1][['x']],d[1][['y']], c=d[1][['c']], s=10, alpha=0.3, cmap='viridis')
    clb = plt.colorbar();
    plt.show()

Returns this error: ValueError: RGBA values should be within 0-1 range

Answer 1

Try this:

df = pd.DataFrame(data)

for d in df.groupby(df['id']):
    plt.plot(d[1][['x','y']])
    plt.show()