使用函数在 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)