to_csv dtype
时间: 2024-06-19 09:02:27 浏览: 149
pandas.read_csv参数详解(小结)
to_csv是pandas库中DataFrame对象的一个方法,用于将DataFrame对象写入到CSV文件中。to_csv中的dtype参数是可选的,用于指定写入CSV文件时每一列的数据类型。如果不指定dtype参数,则默认将所有列的数据类型视为字符串类型。
dtype参数的具体用法如下:
1. dtype=None(默认值):所有列都视为字符串类型。
2. dtype={'列名1': 数据类型1, '列名2': 数据类型2, ...}:指定每个列的数据类型。其中,列名为字符串类型,数据类型可以是以下任意一个Python数据类型:int、float、bool、datetime、timedelta、object、category。
3. dtype=[数据类型1, 数据类型2, ...]:按照列的顺序依次指定每个列的数据类型。
举个例子:
假设有如下DataFrame对象df:
| name | age | gender |
| ---- | --- | ------ |
| Tom | 18 | M |
| Jerry| 22 | M |
| Lily | 20 | F |
将df写入到CSV文件中,并将age列的数据类型指定为整数型,gender列的数据类型指定为分类型,则代码如下:
```
df.to_csv('test.csv', index=False, dtype={'age': int, 'gender': 'category'})
```
阅读全文