get_dummies()函数文档
时间: 2024-04-14 09:27:29 浏览: 99
`get_dummies()`函数是pandas库中的一个函数,用于将分类变量转换为哑变量/虚拟变量。它的作用是将一个包含分类变量的DataFrame或Series转换为包含虚拟变量的DataFrame。
该函数的语法如下:
```python
pandas.get_dummies(data, prefix=None, prefix_sep='_', dummy_na=False, columns=None, sparse=False, drop_first=False, dtype=None)
```
参数说明:
- `data`:要转换的DataFrame或Series对象。
- `prefix`:虚拟变量列名的前缀。默认为None,表示没有前缀。
- `prefix_sep`:虚拟变量列名的前缀与原始列名的分隔符。默认为'_'。
- `dummy_na`:是否添加一个列来表示缺失值。默认为False,表示不添加。
- `columns`:指定要进行转换的列名列表。如果为None,则对所有分类变量列进行转换。
- `sparse`:是否使用稀疏格式存储结果。默认为False,表示不使用稀疏格式。
- `drop_first`:是否删除第一个虚拟变量列。默认为False,表示不删除。
- `dtype`:指定结果DataFrame的数据类型。
该函数会根据原始数据中的每个分类变量创建一个虚拟变量列,每个分类变量的每个取值对应一个新列,取值为1表示该样本属于该类别,取值为0表示不属于。
示例:
```python
import pandas as pd
df = pd.DataFrame({'A': ['a', 'b', 'a'], 'B': ['b', 'a', 'c']})
dummies = pd.get_dummies(df, prefix='dummy', prefix_sep='_')
print(dummies)
```
输出结果:
```
dummy_a dummy_b dummy_c
0 1 0 0
1 0 1 0
2 1 0 0
```
以上是对`get_dummies()`函数的简要说明,更详细的信息可以参考pandas官方文档。
阅读全文