python apply函数

import pandas as pd

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

# 示例1: 使用apply函数对每一列进行操作
# 这里我们对每一列求和
result = df.apply(sum)
print("对每一列求和的结果：")
print(result)

# 示例2: 使用apply函数对每一行进行操作
# 这里我们对每一行求和，并传入额外参数axis=1表示按行操作
result_row = df.apply(sum, axis=1)
print("\n对每一行求和的结果：")
print(result_row)

# 示例3: 使用自定义函数与apply结合
def custom_function(x):
    return x * 2

# 对每一列应用自定义函数
result_custom = df.apply(custom_function)
print("\n使用自定义函数对每一列操作的结果：")
print(result_custom)

解释说明：

• apply 函数是 Pandas 中非常常用的一个方法，它可以对 DataFrame 或 Series 的每一行或每一列应用某个函数。
• 在示例1中，apply(sum) 对每一列求和，默认情况下 apply 是按列操作的（axis=0）。
• 在示例2中，通过传递 axis=1 参数，apply 变为按行操作，因此对每一行求和。
• 在示例3中，我们定义了一个自定义函数 custom_function，并通过 apply 将其应用于每一列，实现了对每一列元素的加倍操作。