Я пытаюсь написать функцию для использования с стилем pandas. Я хочу, чтобы высокие столбцы указывались в аргументах. Это не очень элегантно, но, например:
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']))
Я получаю сообщение об ошибке: ( "Объект" Серии "не имеет атрибутов" столбцы ")
Я думаю, что я принципиально не совсем понимаю, как стилист вызывает и применяет эту функцию.
Спасибо,