我試圖在列中繪制值，并通過從外部手動傳遞顏色來根據(jù)不同列中的組對它們進行著色。我將藍色分配給測試組，將紅色分配給訓練組。但以下代碼僅以藍色繪制值。import datetimeimport pandas as pddft2 = pd.DataFrame({'A': [1,3,3,4,1,3,3,4], 'B': [2015, 2016, 2017,2018, 2019, 2020, 2021, 2022], 'C': pd.Categorical(["test", "train", "test", "train","train","train","train","train"])})dft2 = dft2.set_index('B')colors = {'test': 'b', 'train': 'r'}dft2['A'].plot(figsize=(24,3), rot=90, color=[colors[i] for i in dft2['C']])
1 回答
www說
TA貢獻1775條經(jīng)驗 獲得超8個贊
您可能需要散點圖版本
import datetime
import pandas as pd
dft2 = pd.DataFrame({'A': [1,3,3,4],
'B': [2015, 2016, 2017,2018],
'C': pd.Series(1, index=list(range(4)), dtype='float32'),
'D': np.array([3] * 4, dtype='int32'),
'E': pd.Categorical(["test", "train", "test", "train"]),
'F': 'foo'})
dft2.plot.scatter(x='B',y='A',c=['b' if i== 'test' else 'r' for i in dft2['E']])
添加回答
舉報
0/150
提交
取消