如何将函数逐元素应用到多个 Pandas DataFrame 列？

Pandas：将函数应用于多列

问题：

给定一个函数和一个具有多列的 Pandas DataFrame，我们如何将函数逐元素应用到数组中的每对对应值列？

解决方案：

有一种有效的方法可以使用 Pandas 的 lambda 函数和 apply 方法来实现此目的：

def get_sublist(sta, end):
    return mylist[sta:end+1]

df['col_3'] = df.apply(lambda x: get_sublist(x.col_1, x.col_2), axis=1)

这适用get_sublist 函数添加到 DataFrame 的每一行，使用“col_1”和“col_2”列中的值作为输入。结果是一个新列“col_3”，其中包含计算的子列表。

用法：

考虑以下示例：

df = pd.DataFrame({'ID':['1','2','3'], 'col_1': [0,2,3], 'col_2':[1,4,5]})
mylist = ['a','b','c','d','e','f']

# Apply the function and create a new column
df['col_3'] = df.apply(lambda x: get_sublist(x.col_1, x.col_2), axis=1)

print(df)

输出：

  ID  col_1  col_2            col_3
0  1      0      1       ['a', 'b']
1  2      2      4  ['c', 'd', 'e']
2  3      3      5  ['d', 'e', 'f']

以上是如何将函数逐元素应用到多个 Pandas DataFrame 列？的详细内容。更多信息请关注PHP中文网其他相关文章！