如何使用列表列表为熊猫数据框设置样式？

Question

我有一个熊猫数据框，如：

我需要使用类似列表的样式来设置样式

[[3, 7, 4, 5],
[6, 17, 5, 10, 13, 16],
[7, 22, 6, 17, 19, 12],
[12, 26, 24, 25, 23, 18, 20],
[21, 20, 18, 27, 25]]

如果R1值在第一列表中为蓝色，如果R2值在第二列表中为蓝色，依此类推。 换句话说，如果值在对应的列表中，则每列的颜色编号。

我尝试过：

def posclass(val):

    color = 'black'

    for i in range(5):
    
        if (val in list[i]):

            color = 'blue'

    return 'color: %s' % color

df.style.applymap(posclass,  subset=['R1','R2','R3','R4','R5'])

但是将每个列表应用于每列都无法正常工作。

期望的结果是一个带有彩色数字的数据框（与每个列表的每一列都匹配的数字）。

Answer 1

尝试这样的事情：

df = pd.DataFrame(np.arange(40).reshape(-1,4), columns=[f'R{i}' for i in range(1,5)])

输入df：

   R1  R2  R3  R4
0   0   1   2   3
1   4   5   6   7
2   8   9  10  11
3  12  13  14  15
4  16  17  18  19
5  20  21  22  23
6  24  25  26  27
7  28  29  30  31
8  32  33  34  35
9  36  37  38  39

和

list_l = [[3, 7, 4, 5],
[6, 17, 5, 10, 13, 16],
[7, 22, 6, 17, 19, 12],
[12, 26, 24, 25, 23, 18, 20],
[21, 20, 18, 27, 25]]

然后：

def f(x):
    colpos = df.columns.get_loc(x.name)
    return ['color: blue' if n in list_l[colpos] else '' for n in x]
df.style.apply(f)

输出：