学堂 学堂 学堂公众号手机端

Pandas中怎么应用自定义函数

lewis 4年前 (2021-12-18) 阅读数 6 #技术

在Pandas中应用自定义函数通常使用apply方法。下面是一个简单的示例:

假设有一个包含一列数字的DataFrame:

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3, 4, 5]})

现在我们定义一个自定义函数,例如计算数字的平方:


def square(x):
    return x ** 2

然后我们可以使用apply方法将这个函数应用到DataFrame的列上:

df['B'] = df['A'].apply(square)

这样就会得到一个新的列’B’,里面存放的是’A’列中每个数字的平方值。

版权声明

本文仅代表作者观点,不代表博信信息网立场。

热门