파이썬 사전 값을 기반으로 pandas 데이터 프레임의 개별 셀에 색상을 지정하는 방법

다음 팬더 df가 있습니다.

      Player    Team     EPA
0   L.Jackson   BAL     0.33
1   P.Mahomes   KC      0.25
2   D.Brees     NO      0.24
3   M.Stafford  DET     0.21
4   D.Prescott  DAL     0.19
5   R.Tannehill TEN     0.18

키가 Team 필드와 일치하는 다음 사전 값을 적용하여 색상을 사용하여 스타일을 지정하고 싶습니다. 또한 Player 필드의 해당 값을 동일한 색상으로 만들고 싶습니다.

COLORS = {'BAL':'#241773','DAL':'#B0B7BC','DET':'#046EB4',
          'KC':'#CA2430','NO':'#A08A58','TEN':'#4095D1'}

def highlight_cols(s, coldict):
    if qbs.Team in COLORS.keys():
        return ['background-color: {}'.format(COLORS[qbs.Team])] * len(s)
    return [''] * len(s)

qbs.style.apply(highlight_cols, coldict=COLORS)

 

해결 방법

 

EPA 열에 만 색상을 지정 하려는지 확실하지 않고 전체 행을 원하는 경우 마지막 줄에서 subset 인수를 제거합니다.

def highlight_cols(s, coldict):
    return ['background-color: {}'.format(COLORS[v]) if v else '' for v in qbs.Team.isin(COLORS.keys())*qbs.Team.values]
    

qbs.style.apply(highlight_cols, coldict=COLORS, subset='EPA')

 

참조 페이지 https://stackoverflow.com/questions/63759891