python 对列数据进行打标签
时间: 2023-12-04 15:04:30 浏览: 73
可以使用 pandas 库对列数据进行打标签。
例如,假设有一个名为 data 的 DataFrame,其中包含一个名为 age 的列,我们可以使用以下代码将该列的数据划分为不同的年龄段并打上标签:
```
import pandas as pd
# 定义年龄段和对应的标签
bins = [0, 18, 30, 50, 100]
labels = ['青少年', '年轻人', '中年人', '老年人']
# 读取数据
data = pd.read_csv('data.csv')
# 对 age 列进行分段并打上标签
data['age_group'] = pd.cut(data['age'], bins=bins, labels=labels)
# 输出结果
print(data.head())
```
上述代码中,我们首先定义了年龄段和对应的标签,然后使用 pandas 的 cut 函数将 age 列的数据分成不同的年龄段,并将结果存储在一个新的列 age_group 中。最后,我们输出了 DataFrame 的前几行,可以看到新的列已经添加成功。
阅读全文