python中apply函数的用法

作者：步崖发布日期:2025-03-25 浏览:147

import pandas as pd

# 创建一个简单的 DataFrame
data = {'A': [1, 2, 3], 'B': [4, 5, 6]}
df = pd.DataFrame(data)

# 示例1: 使用 apply 对每一列应用函数
# 这里我们对每一列的元素求平方
result1 = df.apply(lambda x: x ** 2)
print("对每一列应用函数 (求平方):")
print(result1)

# 示例2: 使用 apply 对每一行应用函数
# 这里我们计算每一行的和
result2 = df.apply(lambda x: x.sum(), axis=1)
print("\n对每一行应用函数 (求和):")
print(result2)

# 示例3: 使用 apply 对特定列应用函数
# 这里我们只对列 'A' 的元素求平方
result3 = df['A'].apply(lambda x: x ** 2)
print("\n对特定列应用函数 (对列 'A' 求平方):")
print(result3)

解释说明：

apply 函数是 Pandas 中非常常用的工具，可以对 DataFrame 或 Series 的行或列应用自定义函数。
axis=0 表示按列操作，默认行为；axis=1 表示按行操作。
在示例1中，我们使用 lambda 函数对每一列的元素求平方。
在示例2中，我们计算每一行的和。
在示例3中，我们只对特定列 'A' 的元素求平方。

上一篇：python 文件夹下的文件名

下一篇：python translate