Pandas Apply Lambda表达式来处理数据|极客教程

Pandas Apply Lambda表达式来处理数据

在数据分析过程中，经常需要对数据进行复杂的转换和计算。Pandas 是一个强大的 Python 数据分析库，它提供了许多方法来处理和分析数据。其中，apply 函数是一个非常有用的工具，它允许用户对 DataFrame 或 Series 中的数据应用一个函数。当结合使用 lambda 匿名函数时，apply 可以非常灵活地处理各种数据操作，无需定义复杂的函数。

本文将详细介绍如何在 Pandas 中使用 apply 函数结合 lambda 表达式来处理数据，包括多个实用的示例代码。

1. Pandas DataFrame 和 Series 简介

在深入了解 apply 和 lambda 的使用之前，首先需要了解 Pandas 中的两个基本数据结构：DataFrame 和 Series。

DataFrame 是一个二维标签数据结构，你可以把它想象成一个表格，或者是一个 SQL 表，或者是一个字典类型的对象。它是由多个列组成的，每一列可以是不同的值类型（数值、字符串、布尔值等）。
Series 是一个一维标签数组，可以包含任何数据类型（整数、字符串、浮点数、Python 对象等）。Series 可以从列表、数组或单个数据生成。

2. 使用 apply 函数

apply 函数可以被用于 Series 和 DataFrame。在 Series 上，它通常用于应用一个函数到每个元素上。在 DataFrame 上，它可以被用来应用一个函数到每行或每列上。

示例代码 1：对 Series 应用 lambda 函数

import pandas as pd

# 创建一个 Series
s = pd.Series([20, 21, 12], index=['pandasdataframe.com', 'b', 'c'])

# 使用 apply 和 lambda 函数将每个数字增加 10
result = s.apply(lambda x: x + 10)
print(result)

Output:

Pandas Apply Lambda表达式来处理数据

示例代码 2：对 DataFrame 的每一列应用 lambda 函数

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
}, index=['pandasdataframe.com', 'b', 'c'])

# 使用 apply 和 lambda 函数将每个元素乘以 2
result = df.apply(lambda x: x * 2)
print(result)

Output:

Pandas Apply Lambda表达式来处理数据

示例代码 3：对 DataFrame 的每一行应用 lambda 函数

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
}, index=['pandasdataframe.com', 'b', 'c'])

# 使用 apply 和 lambda 函数计算每行的和
result = df.apply(lambda x: x.sum(), axis=1)
print(result)

Output:

Pandas Apply Lambda表达式来处理数据

3. apply 和 lambda 结合使用的高级示例

示例代码 4：使用条件逻辑

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'A': [10, 20, 30],
    'B': [20, 30, 40]
}, index=['pandasdataframe.com', 'b', 'c'])

# 使用 apply 和 lambda 函数应用条件逻辑
result = df.apply(lambda x: x[x > 15])
print(result)

Output:

Pandas Apply Lambda表达式来处理数据

示例代码 5：修改 DataFrame 的某一列

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
}, index=['pandasdataframe.com', 'b', 'c'])

# 使用 apply 和 lambda 函数修改列 'A'
df['A'] = df['A'].apply(lambda x: x ** 2)
print(df)

Output:

Pandas Apply Lambda表达式来处理数据

示例代码 6：结合使用多个列

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
}, index=['pandasdataframe.com', 'b', 'c'])

# 使用 apply 和 lambda 函数结合多个列
result = df.apply(lambda row: row['A'] + row['B'], axis=1)
print(result)

Output:

Pandas Apply Lambda表达式来处理数据

示例代码 7：使用 apply 处理文本数据

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'Text': ['hello', 'world', 'pandasdataframe.com']
}, index=['a', 'b', 'c'])

# 使用 apply 和 lambda 函数转换文本为大写
result = df['Text'].apply(lambda x: x.upper())
print(result)

Output:

Pandas Apply Lambda表达式来处理数据

示例代码 8：使用 apply 进行日期转换

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'Date': ['2021-01-01', '2021-02-01', '2021-03-01']
}, index=['pandasdataframe.com', 'b', 'c'])

# 使用 apply 和 lambda 函数将字符串日期转换为 datetime 对象
df['Date'] = pd.to_datetime(df['Date'].apply(lambda x: x))
print(df)

Output:

Pandas Apply Lambda表达式来处理数据

示例代码 9：过滤 DataFrame

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3, 4, 5],
    'B': [5, 4, 3, 2, 1]
}, index=['pandasdataframe.com', 'b', 'c', 'd', 'e'])

# 使用 apply 和 lambda 函数过滤数据
result = df.apply(lambda x: x[x > 3])
print(result)

Output:

Pandas Apply Lambda表达式来处理数据

示例代码 10：对 DataFrame 的特定行进行操作

import pandas as pd

# 创建一个 DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6]
}, index=['pandasdataframe.com', 'b', 'c'])

# 使用 apply 和 lambda 函数对特定行进行操作
result = df.loc['pandasdataframe.com'].apply(lambda x: x * 2)
print(result)