python - 如何使用字典设置 Python Pandas Dataframe 的样式

Question

我有以下索引为“NUM”的 Pandas 数据框:

我想使用以下字典为单元格添加颜色:

d =  {'R1': [1,2,3,4,5,6,7,8,9,10]
  ,'R2': [3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20]
  ,'R3': [7,8,9,10,11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26]
  ,'R4': [11,12,13,14,15,16,17,18,19,20,21,22,23,24,25,26,27]
  ,'R5': [17,18,19,20,21,22,23,24,25,26,27,28]
}

字典值(列表)中的值是数据框中的索引。 我想为单元格添加背景颜色，例如“R1”列颜色 1 到 10，“R2”列颜色 3 到 20 等等。

谢谢

Answer 1

你可以试试:

df.style.apply(
    lambda x: ['background-color:red' if i in d[x.name] else '' 
               for i in np.arange(len(x))+1]
)

输出: