我一直在尝试编写一个与 pandas 一起使用的函数style
。我想突出显示我在参数中指定的特定列。这不是很优雅,但是例如:
data = pd.DataFrame(np.random.randn(5, 3), columns=list('ABC'))
def highlight_cols(df, cols, colcolor = 'gray'):
for col in cols:
for dfcol in df.columns:
if col == cols:
color = colcolor
return ['background-color: %s' % color]*df.shape[0]
然后调用:
data.style.apply(highlight_cols(cols=['B','C']))
我收到一个错误:
'Series' object has no attribute 'columns'
我想我从根本上不太了解样式器是如何调用apply
函数的。