為每個唯一ID創建一個圖

Question

給定數據幀df，我可以使用一些幫助來創建兩個不同的散點圖，一個用於x，y坐標，c值用於ID為“ aa”的顏色映射，另一個用於x，y坐標，c值用於ID為“ bb”的顏色圖。 實際數據中有超過1000個唯一ID。

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()

返回此錯誤：ValueError：RGBA值應在0-1范圍內

Answer 1

嘗試這個：

df = pd.DataFrame(data)

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