[PYTHON] 특정 그룹에 해당하는 bar 들만 색칠하기

0. 목표

 - 특정 그룹에 해당하는 bar 들만 색칠하기

1. 실습

 1) library 호출

import pandas as pd

 2) 데이터 생성

df = pd.DataFrame({'x' : ['a', 'b', 'c', 'd', 'e'], 'y' : [2, 7, 3, 9, 4]})

 3) MultiIndex

index = [['A', 'A', 'A', 'B', 'B'], [1, 2, 3, 4, 5]]
df.index = index

 4) 그림그리기

fig, ax = plt.subplots()
fig.set_size_inches(15, 10)

df.sort_values('y', ascending = True, inplace = True)
df.reset_index(inplace = True)

bars = plt.barh(df['x'], df['y'])
for i in df[df['level_0'] == 'A'].index:
    bars[i].set_color('orange')
    
for i in df[df['level_0'] == 'B'].index:
    bars[i].set_color('red')
    
ax.tick_params(axis = 'x', labelsize = 20)
ax.tick_params(axis = 'y', labelsize = 20)

2. 참고

 - MultiIndex / advanced indexing

 - 멀티인덱스 생성