pandas dataframe subplot grouping by columns

Question

df = pd.DataFrame([[0, 1, 2], [0, 1, 2]])
df.plot(subplots=True)

I want subplot by group [0, 1] and [2] columns. is there the way?

Answer 1

You can use DataFrameGroupBy.plot by Index.map by dictionary for 2 groups:

mapping = {0:'a', 1:'a', 2:'b'}

df.groupby(df.columns.map(mapping.get), axis=1).plot()

Detail:

print (df.columns.map(mapping.get))
Index(['a', 'a', 'b'], dtype='object')