Pandas里怎样创建自定义函数

在Pandas中，您可以使用.apply()方法将自定义函数应用于DataFrame的列或行。以下是创建和使用自定义函数的步骤：

1. 首先，导入所需的库：

import pandas as pd

2. 创建一个示例DataFrame：

data = {'A': [1, 2, 3, 4],
        'B': [5, 6, 7, 8],
        'C': [9, 10, 11, 12]}

df = pd.DataFrame(data)

3. 定义一个自定义函数。例如，我们创建一个函数，将输入值乘以2：

def multiply_by_two(x):
    return x * 2

4. 使用.apply()方法将自定义函数应用于DataFrame的列或行。例如，将自定义函数应用于列’A’：

df['A'] = df['A'].apply(multiply_by_two)

现在，DataFrame df 的列 ‘A’ 已经被更新，所有值都乘以了2。

您还可以将自定义函数应用于整个DataFrame，而不是单个列。在这种情况下，您需要使用axis=1参数：

def custom_function(row):
    # 在这里处理整行数据
    return row['A'] + row['B']

df['D'] = df.apply(custom_function, axis=1)

在这个例子中，我们创建了一个名为custom_function的自定义函数，它接受一行数据作为输入，并返回列’A’和列’B’的和。然后，我们使用.apply()方法将这个函数应用于整个DataFrame，并将结果存储在新列’D’中。