在具有数值的列上的pandas数据框上逐行应用函数

pdu*_*ois 6 python pandas

我有以下数据框:

import pandas as pd
df = pd.DataFrame({'AAA' : ['w','x','y','z'], 'BBB' : [10,20,30,40],'CCC' : [100,50,-30,-50]})
Run Code Online (Sandbox Code Playgroud)

看起来像这样:

In [32]: df
Out[32]:
  AAA  BBB  CCC
0   w   10  100
1   x   20   50
2   y   30  -30
3   z   40  -50
Run Code Online (Sandbox Code Playgroud)

我想要做的是对除了具有非数值的那些列(在这种情况下AAA)之外的每一列执行函数操作.在实际情况下,非数字情况总是在第一列,其余(可能大于2列)总是数字.

最终的期望输出是:

  AAA  BBB  CCC  Score
0   w   10  100  110
1   x   20   50   70
2   y   30  -30    0
3   z   40  -50  -10
Run Code Online (Sandbox Code Playgroud)

我尝试过但失败了:

import numpy as np
df["Score"] = df.apply(np.sum, axis=1)
Run Code Online (Sandbox Code Playgroud)

什么是正确的方法呢?

UPDATE2:

这是给出的代码SettingWithCopyWarning.请重新开始ipython测试.

import pandas as pd
import numpy as np 
def cvscore(fclist):
    sd = np.std(fclist)
    mean = np.mean(fclist)
    cv = sd/mean
    return cv

def calc_cvscore_on_df(df):
    df["CV"] = df.iloc[:,1:].apply(cvscore, axis=1)
    return df

df3 = pd.DataFrame(np.random.randn(1000, 3), columns=['a', 'b', 'c'])
calc_cvscore_on_df(df3[["a","b"]])
Run Code Online (Sandbox Code Playgroud)

unu*_*tbu 12

要选择除第一列之外的所有内容,您可以使用df.iloc[:, 1:]:

In [371]: df['Score'] = df.iloc[:, 1:].sum(axis=1)

In [372]: df
Out[372]: 
  AAA  BBB  CCC  Score
0   w   10  100    110
1   x   20   50     70
2   y   30  -30      0
3   z   40  -50    -10
Run Code Online (Sandbox Code Playgroud)

func对每行应用任意函数:

df.iloc[:, 1:].apply(func, axis=1)
Run Code Online (Sandbox Code Playgroud)

例如,

import numpy as np
import pandas as pd

def cvscore(fclist):
    sd = np.std(fclist)
    mean = np.mean(fclist)
    cv = sd/mean
    return cv

df = pd.DataFrame({'AAA' : ['w','x','y','z'], 'BBB' : [10,20,30,40],
                   'CCC' : [100,50,-30,-50]})

df['Score'] = df.iloc[:, 1:].apply(cvscore, axis=1)
print(df)
Run Code Online (Sandbox Code Playgroud)

产量

  AAA  BBB  CCC     Score
0   w   10  100  1.211386
1   x   20   50  0.868377
2   y   30  -30       NaN
3   z   40  -50 -5.809058
Run Code Online (Sandbox Code Playgroud)

  • @pdubois:警告说'df3 [['a','b']] ['CV'] = ...`将无法修改`df3`.由于`calc_cvscore_on_df`返回一个新的`df`,看起来你并没有试图修改`df3`.因此,避免警告的最简单方法是将`df = df.copy()`放在`calc_cvscore_on_df`的开头.那么`df`将是它自己的独立DataFrame,`df ['CV'] = ...`不会引发警告. (2认同)