使用python在数据框列中用逗号分隔单元格中的值求和

Question

我有一个数据框列，其中的数值由逗号（，）分隔。我正在尝试查找该列的每个单元格中的值的总和并创建一个新列。

输入： df：

       id Scores
       1  10,12
       2  11
       3  1,2,3

需要输出：

      id Scores Sum
      1  10,12   22
      2  11      11 
      3  1,2,3   6

代码：我试过写一个自定义函数，这样我就可以在python中使用apply函数了

def sum_of_number(listx):
    a=[]
    n=0
    if len(listx)==1:
        a=listx
        n=a
    else:
        a=str(listx).split(',')
        for i in range(0,len(a)):
            n=n+int(a[i])
    return n

df['Sum']=df.Scores.apply(sum_of_number)

但我无法获得预期的结果。

---

编辑： 感谢 DeepSpace 提供解决方案

df['Sum'] = df['Scores'].apply(lambda x: sum(map(float, x.split(','))))

同样可以得到中位数

      id Scores Median
      1  10,12   11
      2  11      11 
      3  1,2,3   2

Answer 1

您可以通过组合sum、map 和int（或float，根据您的需要）使用更简单的代码：

import pandas as pd

df = pd.DataFrame({'Scores': ['10,12', '11', '1,2,3']})
df['Sum'] = df['Scores'].apply(lambda x: sum(map(int, x.split(','))))
# or df['Sum'] = df['Scores'].apply(lambda x: sum(map(float, x.split(','))))
print(df)

#    Scores  Sum
#  0  10,12   22
#  1  11      11
#  2  1,2,3    6

以上假设值是字符串。如果这些值是整数的实际列表，那就更容易了：

import pandas as pd

df = pd.DataFrame({'Scores': [[10, 12], [11], [1, 2, 3]]})
df['Sum'] = df['Scores'].apply(sum)
print(df)

#        Scores  Sum
#  0   [10, 12]   22
#  1       [11]   11
#  2  [1, 2, 3]    6