将第一列数据类型从float转换为int，然后写回原始csv文件

Question

我想将第一列的数据类型从float更改为integer，并将整数存储在原始文件中。这是我的一些代码。

import csv 
with open("data.csv", r) as f: 
csvreader = csv.reader(f) 
for row in csvreader: 
i = int(row[0]) # first column of the row 

然后我不知道如何修改这个步骤后的原始csv文件。谢谢你的帮助。

Answer 1

使用pandas:可以很容易地做到这一点

import pandas as pd
df = pd.read_csv("data.csv")
df["col"] = df["col"].astype(int)
df.to_csv("data.csv", index=False)

如果您不知道第一列的名称，并且希望使用基于整数的索引，那么请执行以下操作：

df.iloc[:, 0] = df.iloc[:, 0].astype(int)