【问题标题】:Multi Conditional if Statement Pandas (code doesn't finish running)多条件 if 语句 Pandas(代码未完成运行)
【发布时间】:2022-12-18 03:09:05
【问题描述】:

尝试根据其他列值在 pandas 中创建多条件新列。

以下代码不会产生错误或结果(但会产生警告);它一直在运行:

for val1,val2 in zip(df['a'], df['b']):
    if val1 == 0 and val2 == 0:
        df['new_column'] = 0
    elif val1 in df['a'] == 0:
      
        df['new_column'] = 1
    else:
        for val2 in df['b']:
            if val2 ==0:
                df['new_column'] = 0
            else:
                df['new_column'] = df['b'] /  df['a']

df 看起来像:

['a']    ['b']
  0        0  
  0      1000
1000       0
5000     2000

期望 df['new column'] 是这样的:

['new column']
      0
      1
      0
     .4

【问题讨论】:

  • 请提供输入/预期输出的最小可重现示例,以及逻辑
  • 我不能说我真的很喜欢在这里的外循环和内循环中都使用val2 作为变量名。那可能是一个错误。
  • 您的代码中存在语法错误。但主要问题是无法以这种方式形成 DataFrame 列。查看指明方向的答案。

标签: python pandas


【解决方案1】:

这可能不是最优雅的解决方案,但根据问题中提供的有限信息,此代码至少会生成您预期的输出:

def myfunc(row):
    if row['a'] == 0 and row['b'] == 0:
        result = 0
    else:
        if row['a'] == 0:
            result = 1
        elif row['b'] == 0:
            result = 0
        else:
            result = row['b'] / row['a']
    return result

df['new column'] = df.apply(myfunc, axis=1)

【讨论】:

    【解决方案2】:

    与@cucurbit 在细节上有所不同,但原理相同:

    def vals(x, y):
        if x == 0 and y == 0:
            return 0
        if x == 0 :
            return 1
        if y == 0:
            return 0
        else:
            return y/x
    
    df['new column'] = df.apply(lambda x : vals(x['a'], x['b']), axis =1)
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2022-06-14
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2012-11-27
      相关资源
      最近更新 更多