dummies函数怎么用
时间: 2023-08-13 13:07:47 浏览: 36
`dummies` 函数通常用于将分类变量转换成虚拟变量。它可以通过将分类变量转换为二进制形式来实现。
下面是使用 `pandas` 库的 `get_dummies()` 函数将分类变量转换为虚拟变量的例子:
```python
import pandas as pd
# 创建一个包含分类变量的数据框
df = pd.DataFrame({'Animal': ['cat', 'dog', 'dog', 'cat', 'bird']})
# 使用 get_dummies() 函数转换分类变量
dummies = pd.get_dummies(df['Animal'])
# 将虚拟变量添加回原始数据框
df = pd.concat([df, dummies], axis=1)
# 打印转换后的数据框
print(df)
```
输出结果:
```
Animal bird cat dog
0 cat 0 1 0
1 dog 0 0 1
2 dog 0 0 1
3 cat 0 1 0
4 bird 1 0 0
```
在上面的例子中,我们首先使用 `pandas` 库的 `DataFrame` 函数创建一个包含分类变量的数据框。然后,使用 `get_dummies()` 函数将分类变量 `Animal` 转换为虚拟变量。最后,将虚拟变量添加回原始数据框,并打印转换后的数据框。
注意,`get_dummies()` 函数还可以处理多个分类变量,并且可以指定前缀和分隔符来更改虚拟变量的列名。