Условное моделирование в пандах с использованием других столбцов

Question

Я искал и не могу найти ответ на этот вопрос в любом месте, так что, надеюсь, это возможно. У меня есть датафрейм, для простоты я приведу сокращенную версию ниже. То, что я хотел бы сделать, это применить пользовательскую формулу для стиля или стилизовать один конкретный столбец на основе значений в другом столбце.

Используя это в качестве примера, я хотел бы выделить ячейки текущего столбца, где Diff> Historic Standard Dev в этой строке.

Я изучил подходы style.apply, но не могу найти подходящего. Любая помощь будет принята с благодарностью.

Спасибо!

Answer 1

Вы можете создать DataFrame стилей с помощью Styler.apply:

def select_col(x):
    c1 = 'background-color: red'
    c2 = '' 
    #compare columns
    mask = x['Diff'] > x['HistoricStandardDev']
    #DataFrame with same index and columns names as original filled empty strings
    df1 =  pd.DataFrame(c2, index=x.index, columns=x.columns)
    #modify values of df1 column by boolean mask
    df1.loc[mask, 'Current'] = c1
    return df1

df.style.apply(select_col, axis=None)