Matplotlib線圖從數據透視表：行

Question

的自定義色彩鑑於以下幾點：在t.plot

import pandas as pd 
import numpy as np 
%matplotlib inline 
df = pd.DataFrame(
     {'YYYYMM':[201603,201503,201403,201303,201603,201503,201403,201303], 
     'Count':[5,6,2,7,4,7,8,9], 
     'Group':['A','A','A','A','B','B','B','B']}) 
df['YYYYMM']=df['YYYYMM'].astype(str).str[:-2] 
t=df.pivot_table(df,index=['YYYYMM'],columns=['Group'],aggfunc=np.sum) 

fig, ax = plt.subplots(1,1) 
t.plot(ax=ax) 

是否有參數（），讓我可以指定每行的顏色？

提前感謝！

Answer 1

您可以提供的線條樣式：

t.plot(ax=ax, style=['yellow', 'red']) 

Answer 2

您可以使用：

ax.set_color_cycle(['red', 'black']) 

樣品：

df = pd.DataFrame(
     {'YYYYMM':[201603,201503,201403,201303,201603,201503,201403,201303], 
     'Count':[5,6,2,7,4,7,8,9], 
     'Group':['A','A','A','A','B','B','B','B']}) 
df['YYYYMM']=df['YYYYMM'].astype(str).str[:-2] 
t=df.pivot_table(df,index=['YYYYMM'],columns=['Group'],aggfunc=np.sum) 

fig, ax = plt.subplots(1,1) 
ax.set_color_cycle(['red', 'black']) 
t.plot(ax=ax) 

編輯：

非常有趣，通過它似乎更是用顏色的全名，因爲它與Mike 1. answer：

t.plot(ax=ax, style=['yellow', 'red'])