apply函数用法

apply函数是pandas库中的一个函数，用于在DataFrame或Series的行或列上应用一个函数。

语法：

DataFrame.apply(func, axis=0, raw=False, result_type=None, args=(), **kwds)

Series.apply(func, convert_dtype=True, args=(), **kwds)

参数说明：

func：需要应用的函数。

axis：轴向，0表示按列应用，1表示按行应用。

raw：默认为False，表示传递的函数func的参数是一个序列。

result_type：指定返回结果的类型，默认为None，表示返回一个与原DataFrame或Series结构相同的对象。

args：传递给func的其他参数。

示例：

1. 对DataFrame的每一列求和

import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df.apply(sum, axis=0)

输出：

A    6
B    15
dtype: int64

2. 对DataFrame的每一行求和

import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df.apply(sum, axis=1)

输出：

0     5
1     7
2     9
dtype: int64

3. 对Series的每个元素求平方根

import pandas as pd
s = pd.Series([1, 4, 9])
s.apply(lambda x: x ** 0.5)

输出：

0    1.0
1    2.0
2    3.0
dtype: float64