代码之家 › 专栏 › 技术社区 › Mainland

Python Dataframe添加两列,其中包含列表

pandas-apply lambda dataframe pandas python

Mainland · 技术社区 · 3 年前

我有一个由两列组成的数据框架,每列都包含列表元素。我想对两个列表执行行元素加法。我下面的解决方案就是从这个答案中得到启发的 Element-wise addition of 2 lists?

我的代码:

df = pd.DataFrame({'A':[[1,2,3],[10,20]],'B':[[4,5,6],[40,50]]})

           A          B
0  [1, 2, 3]  [4, 5, 6]
1   [10, 20]   [40, 50]

df['C'] = df[['A','B']].apply(lambda x: list(map(add,x[0],x[1])))
df['C'] = 
    A   B
0  11  44
1  22  55

预期答案:

df['C'] = 
   C
0 [5,7,9]
1 [50,70]

1 回复 | 直到 3 年前

enke 3 年前

因为需要按行应用函数,所以只需要 axis=1 :

from operator import add
df['C'] = df[['A','B']].apply(lambda x: list(map(add,x[0],x[1])), axis=1)

另一种选择是 explode 名单; sum ; 然后 groupby.agg 要重新获取列表,请执行以下操作:

df['C'] = df.explode(['A','B']).sum(axis=1).astype(int).groupby(level=0).agg(list)

输出:

           A          B          C
0  [1, 2, 3]  [4, 5, 6]  [5, 7, 9]
1   [10, 20]   [40, 50]   [50, 70]

推荐文章

Codeforces Fan · 如何在c中的STL集合中添加lambda函数或执行自定义操作++

3 年前

Amit Saha · Java Stream iterate在检查奇偶时不使用谓词

3 年前

zo Aoo · 当我在lambda函数中更改变量名时,结果会更改[重复]

3 年前

Albert Wijaya · 有没有办法让扩展名按字母顺序排列,而不是按长度排列?

3 年前

Criwran · Kotlin:类型推断失败。类型参数T的值应在输入类型中提及

3 年前

Sergei · 删除数据框中多个列的列表中的某些元素

3 年前

NormalLLer · 在java中,“运算符“<”不能应用于“<lambda参数>”,“int”

3 年前

Schadre · 我如何使用lambda的map函数将下面的列表从摄氏度转换为华氏度?

3 年前

Mainland · Python Dataframe将索引映射到一列列表以提取元素

3 年前

Mainland · Python Dataframe添加两列,其中包含列表

3 年前