pandas 将两个样式器函数同时应用于 Dataframe

8hhllhi2 于 2022-12-10 发布在其他

关注(0)|答案(1)|浏览(175)

下面是我正在使用的示例脚本。我试图同时将两个styler函数应用到一个 Dataframe ，但如您所见，它只调用colors2函数。同时应用两个函数的最佳方式是什么？

import pandas as pd
df = pd.DataFrame(data=[[-100,500,400,0,222,222], [9000,124,0,-147,54,-56],[77,0,110,211,0,222], [111,11,-600,33,0,22,],[213,-124,0,-147,54,-56]])
df.columns = pd.MultiIndex.from_product([['x','y','z'], list('ab')])

def colors(i):
    if i > 0:
        return 'background: red'
    elif i < 0:
        return 'background: green'
    elif i == 0:
        return 'background: yellow'
    else:
        ''
def colors2(i):
    if i < 0:
        return 'background: red'
    elif i > 0:
        return 'background: green'
    elif i == 0:
        return 'background: yellow'
    else:
        ''
idx = pd.IndexSlice
df.style.applymap (colors, subset=pd.IndexSlice[:, idx['x','b']])
df.style.applymap (colors2, subset=pd.IndexSlice[:, idx[:, 'b']])

pandas

来源：https://stackoverflow.com/questions/74673515/applying-two-styler-functions-simultanesouly-to-a-dataframe

1条答案

按热度按时间

htrmnn0y1#

Chain your applymap commands in the desired order (last one prevails):

(df.style
   .applymap(colors2, subset=pd.IndexSlice[:, pd.IndexSlice[:, 'b']])
   .applymap(colors, subset=pd.IndexSlice[:, pd.IndexSlice['x','b']])
 )

Here pd.IndexSlice['x','b']] is more restrictive than pd.IndexSlice[:, 'b'] so we use it last.
Another option could be to use a single function and to decide the color based on the labels inside it.

import numpy as np
def colors(s):
    if s.name[0] == 'x':
        s = s*-1
    return np.sign(s).map({-1: 'background: red', 1: 'background: green', 0: 'background: yellow'})
     
(df.style
   .apply(colors, subset=pd.IndexSlice[:, pd.IndexSlice[:, 'b']])
 )

Output:

赞(0）回复(0）举报 2022-12-10

我来回答

pandas 将两个样式器函数同时应用于 Dataframe

1条答案

相关问题

热门标签

最新问答