小编Thi*_*thy的帖子

如何根据不同列中的值向 Pandas 数据框添加一列?

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

a     b    c    d   
0.7   0.1  0.2  0.3
0.5   0.2  0.2  0.2
Run Code Online (Sandbox Code Playgroud)

我正在编写一些像下面这样的嵌套循环来添加基于这 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
Run Code Online (Sandbox Code Playgroud)

任何人都可以帮助优化代码。

预期输出:

a     b    c …
Run Code Online (Sandbox Code Playgroud)

python dataframe pandas

2
推荐指数
1
解决办法
37
查看次数

标签 统计

dataframe ×1

pandas ×1

python ×1