Pandas 在多列上应用函数

Question

我正在尝试将函数应用于数据框中的每一列，当我尝试仅对单个固定列名执行此操作时，它可以工作。我尝试在每一列上都这样做，但是当我尝试在函数中将列名作为参数传递时，我得到了一个错误。

如何正确传递参数以在数据框上应用函数？

def result(row,c):

     if  row[c] >=0 and row[c] <=1:
         return 'c'

     elif row[c] >1 and row[c] <=2:
         return 'b'

     else:
         return 'a'


cols = list(df.columns.values)

for c in cols

     df[c] = df.apply(result, args = (c),  axis=1)

TypeError: ('result() takes exactly 2 arguments (21 given)', u'occurred at index 0')

输入数据帧格式：

d = {'c1': [1, 2, 1, 0], 'c2': [3, 0, 1, 2]}
df = pd.DataFrame(data=d)
df
      c1   c2
0     1     3
1     2     0
2     1     1
3     0     2

Answer 1

您无需传递列名即可应用。因为您只想检查列的值是否在特定范围内并且应该返回a, b or c。您可以进行以下更改。

def result(val):

     if  0<=val<=1:
         return 'c'
     elif 1<val<=2:
         return 'b'
     return 'a'


cols = list(df.columns.values)

for c in cols

     df[c] = df[c].apply(result)

请注意，这将替换您的列值。

更快的方法是np.select:

import numpy as np
values = ['c', 'b']
for col in df.columns:
    df[col] = np.select([0<=df[col]<=1, 1<df[col]<=2], values, default = 'a')