python - 如何根据不同列中的值将一列添加到 pandas 数据框？

Question

我有一个如下所示的 Pandas 数据框:

a     b    c    d   
0.7   0.1  0.2  0.3
0.5   0.2  0.2  0.2

我正在编写一些如下所示的嵌套循环，以添加基于这 4 列的列结果。

def class_decider(df):
    for i in df['a']:
        if i > 0.6:
            a = "class A"
        elif:
            for j in df['b']:
                if j > 0.2:
                    a = "class B"
                elif:
                    for k in df['c']:
                        if j > 0.15:
                            a = "class C"
                        elif:
                            for l in df['d']:
                                if l > 0.10:
                                    a = "class D"
                                else:
                                    a = "null"
    return a

谁能帮忙优化一下代码。

预期输出:

a     b    c    d     result
0.7   0.1  0.2  0.3   class A
0.5   0.2  0.2  0.2   class C

Answer 1

IIUC，您可以将 a、b、c 和 d 列与 0.6 进行比较, 0.2, 0.15, 0.10 创建一个 bool 掩码，然后使用 idxmax 沿 axis=1 在此掩码上获取第一个列的名称True 值出现在掩码中。

c = ['a', 'b', 'c', 'd']
m = df[c].gt([0.6, 0.2, 0.15, 0.10])
df['Result'] = m.idxmax(1).radd('Class ').mask(~m.any(1), 'Null')

---

     a    b    c    d   Result
0  0.7  0.1  0.2  0.3  Class a
1  0.5  0.2  0.2  0.2  Class c