使用字典创建新列 - Python

在Python中，我们可以使用字典创建新的列。这是一种很方便的方法，可以轻松地向数据框中添加新的列，并使用字典的键值对来为每个观测值分配值。

代码示例

下面是一个使用字典创建新列的示例代码：

import pandas as pd

# 创建数据框
data = {'姓名': ['小明', '小红', '小刚', '小李'], '学科': ['语文', '数学', '英语', '物理'], '成绩': [85, 92, 88, 95]}
df = pd.DataFrame(data)

# 创建字典，用于添加新的列
rank = {'小明': 'B', '小红': 'A', '小刚': 'A', '小李': 'A+'}

# 使用字典创建新列
df['排名'] = df['姓名'].map(rank)

print(df)

输出结果如下：

| | 姓名 | 学科 | 成绩 | 排名 | |---|----|----|-----|-----| | 0 | 小明 | 语文 | 85 | B | | 1 | 小红 | 数学 | 92 | A | | 2 | 小刚 | 英语 | 88 | A | | 3 | 小李 | 物理 | 95 | A+ |

在上面的示例中，我们创建了一个包含姓名、学科和成绩的数据框。然后，我们创建了一个叫做rank的字典，用于为每个学生分配一个排名。最后，我们使用了map()方法将字典中的值映射到df['姓名']列中，并将结果存储在一个新的df['排名']列中。

必要的说明

值得注意的是，我们使用字典来创建新列时，必须保证字典中所有键的值都存在于数据框的列中，否则会报错。同时，在字典中使用的键必须与数据框中一致，否则无法正确映射值。

结论

使用字典创建新列是一种非常方便的方法，能够快速地向数据框中添加列，并进行值映射。在实际应用中，可以根据需要灵活运用，并结合其他方法和函数实现更多自定义的操作。