将数据框转换为列表时如何防止整数转换为浮点数？ [复制]

Question

在将数据框转为列表时如何防止整数转为浮点数？

我有一个包含 5 列数据的 .csv 文件。前四列没有小数点，而最后一列有。

当我使用“pd.read_csv”将此数据导入我的脚本时，数据会正确导入，前 4 个数字为整数，最后一个数字为浮点数，如下所示：

1,1,10,0,1.0

1,1,11,0,0.6

1,1,12,0,0.0

但是我需要将此数据转换为列表，并且当我这样做时，它将所有数字转换为浮点数。我不想要这个。前四个值必须是整数。

这是我当前的代码，在将其转换为列表后，它提供了一个所有数字都是浮点数的列表：

data_file_name = r'C:\Users\username\Desktop\FileName.csv'

data = pd.read_csv(data_file_name)  #<This part works and the data types are correct, the first 4 are integers
data2 = data.values.tolist() #<here is where everything gets converted to a float, even if it was defined as an int in the df.

这会生成一个列表，其中的数据格式如下：

[[1.0, 1.0, 10.0, 0.0, 1.0], [1.0, 1.0, 11.0, 0.0, 0.6], [1.0, 1.0, 12.0, 0.0, 0.0]]

当我需要这样格式化时：

[[1, 1, 10, 0, 1.0], [1, 1, 11, 0, 0.6], [1, 1, 12, 0, 0.0]]

我能做什么？

我试过了：

[int(i,10) for i in data]

但这会返回此错误：

ValueError: invalid literal for int() with base 10: 'Month'

Answer 1

使用dtype 参数来控制数据类型。

pd.read_csv(data_file_name, dtype={0: "int64", 1: "int64", 2: "int64", 3: "int64", 4: "float64"})