我正在使用一个包含客户信息的数据集来学习Python技术。
下面是DataFrame结构(这些是虚构的记录):
import pandas as pd
df1 = pd.DataFrame({'left_name' : ['James', 'Mary', 'John', 'Patricia'],
'left_age' : [30, 37, 30, 35],
'right_name' : ['Robert', 'Jennifer', 'Michael', 'Linda'],
'right_age' : [30, 31, 38, 35]})
print(df1)
left_name left_age right_name right_age
0 James 30 Robert 30
1 Mary 37 Jennifer 31
2 John 30 Michael 38
3 Patricia 35 Linda 35
应用
transpose
方法到df1
,我们得到以下视图:df2 = df1.T
print(df2)
0 1 2 3
left_name James Mary John Patricia
left_age 30 37 30 35
right_name Robert Jennifer Michael Linda
right_age 30 31 38 35
我的目标是对df2
应用一些样式。特别地,
left_name
和right_name
行应该被突出显示为黄色;left_age
和right_age
行应该被突出显示为蓝色。
在发帖之前,我做了一些研究,成功地突出显示了一个子集:
df2.style.set_properties(subset = pd.IndexSlice[['left_name', 'right_name'], :], **{'background-color' : 'yellow'})
问题是我无法将多个样式组合在一起。如果我尝试使用与上面相同的方法为left_age
和right_age
添加额外的蓝色颜色,将会“丢失”之前的样式。
理想情况下,我希望有一个函数,它以df2
作为输入并返回样式化的DataFrame。