python修改csv数据类型
时间: 2023-10-17 15:21:32 浏览: 105
python基础(二)数据
要修改 CSV 文件中的数据类型,可以使用 Python 的内置 csv 模块和 pandas 库。
使用 csv 模块:
```python
import csv
with open('input.csv', 'r') as infile, open('output.csv', 'w') as outfile:
reader = csv.reader(infile)
writer = csv.writer(outfile)
for row in reader:
# 转换数据类型
row[0] = int(row[0])
row[1] = float(row[1])
# 写入输出文件
writer.writerow(row)
```
使用 pandas 库:
```python
import pandas as pd
df = pd.read_csv('input.csv')
# 转换数据类型
df['col1'] = df['col1'].astype(int)
df['col2'] = df['col2'].astype(float)
# 写入输出文件
df.to_csv('output.csv', index=False)
```
以上代码中,假设 CSV 文件中有两列数据,分别为 col1 和 col2。使用 int() 和 float() 函数将 col1 和 col2 中的数据类型转换为整数和浮点数,并将结果写入输出文件中。
阅读全文