pandas 如何在应用styler函数后删除列

pkbketx9  于 2023-05-21  发布在  其他
关注(0)|答案(2)|浏览(152)

如何在应用样式器后删除列?下面是我的style函数:

def highlight_late(x):
        c1 = 'background-color: red'
        #condition
        m = x['price_1'] < x['price_main_x']
        m2 = x['price_2'] < x['price_main_x']
        m3 = x['price_3'] < x['price_main_x']
        #empty DataFrame of styles
        df1 = pd.DataFrame('', index=x.index, columns=x.columns)
    

    
#set column price_2 by condition
df1.loc[m, 'price_1'] = c1
df1.loc[m2, 'price_2'] = c1
df1.loc[m3, 'price_3'] = c1
df1.loc[m, 'url_x'] = c1
df1.loc[m2, 'url_y'] = c1
df1.loc[m3, 'url'] = c1

return df1

下面的方法返回TypeError:“Styler”对象不支持项删除

styles = myDF.style.apply(highlight_late, axis=None)
del styles['price_1']
del styles['price_2']
del styles['price_3']
styles.to_excel('test.xlsx')

我也尝试:

mydf.style.hide_columns(['price_1', 'price_2', 'price_3']).to_excel('test.xlsx')

它不工作,列不会隐藏。
即使来自https://pandas.pydata.org/docs/reference/api/pandas.io.formats.style.Styler.hide_columns.html的这个简单脚本也无法工作

df = pd.DataFrame([[1, 2, 3], [4, 5, 6]], columns=["a", "b", "c"])
df.style.hide_columns(["a", "b"])
df.to_excel('test2.xlsx')
piv4azn7

piv4azn71#

您可以指定要写入Excel的列。

cols = [col for col in styles.columns if col == condition]
df.to_excel('test2.xlsx', columns=cols)
nbysray5

nbysray52#

pandas v1.4.0使用.hideaxis='columns'.hide_columns已弃用。
尝试hide_columns

styles = myDF.style.hide_columns(['price_1', 'price_2', 'price_3']).apply(highlight_late, axis=None)

或删除它们:

styles = myDF.style.hide_columns(['price_1', 'price_2', 'price_3'])
styles = styles.apply(highlight_late, axis=None)

相关问题