Pandas / Pyplot中的散點圖:如何按類別繪制我試圖使用Pandas DataFrame對象在pyplot中制作一個簡單的散點圖,但是想要一種有效的方法來繪制兩個變量,但是有第三列(鍵)指示的符號。我嘗試過使用df.groupby的各種方法,但沒有成功。下面是一個示例df腳本。這會根據'key1'為標記著色,但Id喜歡看到帶有'key1'類別的圖例。我接近了嗎?謝謝。import numpy as npimport pandas as pdimport matplotlib.pyplot as pltdf = 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()
添加回答
舉報
0/150
提交
取消