将值映射到 Pandas 数据框中的数据 - Python

在 Pandas 中，我们可以轻松地将值映射到数据框中的数据。这是一个非常有用的功能，它可以帮助我们将数据转换为更易于理解和处理的形式。

映射单个值

首先，让我们看一下如何将一个单独的值映射到数据框中的数据。假设我们有一个数据框 df，其中包含一个列 gender，该列包含男性和女性的值。我们想将这些值映射为数字 0 和 1，以便更轻松地计算。

import pandas as pd

# 创建 DataFrame
df = pd.DataFrame({'gender': ['male', 'female', 'male', 'female']})

# 映射
df['gender'] = df['gender'].map({'male': 0, 'female': 1})

print(df)

该代码将输出以下数据框：

   gender
0       0
1       1
2       0
3       1

可以看到，男性被映射为 0，女性被映射为 1。

映射多个列

如果我们有多个列需要映射，我们可以使用 applymap() 方法来对整个数据框应用映射。下面是一个示例：

import pandas as pd

# 创建 DataFrame
df = pd.DataFrame({'gender': ['male', 'female', 'male', 'female'],
                   'department': ['HR', 'IT', 'IT', 'HR']})

# 映射
mapper = {'male': 0, 'female': 1, 'HR': 'A', 'IT': 'B'}
df = df.applymap(lambda x: mapper[x])

print(df)

该代码将输出以下数据框：

  gender department
0      0          A
1      1          B
2      0          B
3      1          A

可以看到，gender 和 department 列中的值都被成功映射为相应的数字和字母。

总结

通过 Pandas 的映射功能，我们可以轻松地将数据框中的值映射为我们需要的格式。这是一项非常有用的技能，可以大大简化我们的数据分析和处理工作。