使用函数在 Pandas 中添加一列[重复]

Posted 2023-02-23

技术标签:

【中文标题】使用函数在 Pandas 中添加一列[重复]【英文标题】：Adding a column in Pandas with a function [duplicate] 【发布时间】：2021-01-18 19:18:14 【问题描述】：

假设我有一个这样的数据框：

  columnA  columnB
0   10        90
1   83        17
2   30        21
...

我有这样的功能：

def my_func(a, b):
    value = #do some calculation
    return value

现在我想根据函数的计算为我的数据框获取一个新列 columnC。显然，df["columnC"]= my_func(df["columnA"], df["columnB"]) 不起作用。

如何添加列？

【问题讨论】：

使用熊猫 apply : df['C'] = df.apply(my_func) 这也将熊猫系列传递给我的函数。我上面给出的例子也是同样的问题。你能发布my_func 的作用吗？这只是一些大的 if/return 块 【参考方案1】：

正确的做法是：

df['c'] = df.apply(lambda row: my_func(row['a'], row['b']), axis=1)

【讨论】：

【参考方案2】：

我找到了解决方法。不过这有点hacky...

df["columnC"]=pd.Series([my_func(row["columnA"], row["columnB"])for index, row in df.iterrows()], index=df.index)

【讨论】：

以上是关于使用函数在 Pandas 中添加一列[重复]的主要内容，如果未能解决你的问题，请参考以下文章