python - 将第一列数据类型从float转换为int，并写回原来的csv文件

Question

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

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文件。谢谢您的帮助。

标签： pythoncsvtype-conversion

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)