python - 当列没有值时，pandas CSV to Parquet 数据类型设置不正确

Question

我正在使用 pandas 数据框 read_csv 函数，并且有时列没有值。在这种情况下，使用dtype参数发送的数据类型将被忽略。

import pandas as pd
df = pd.read_csv("example.csv", dtype={"col1": "str", "col2": "float", "col3": "str"})
df.to_parquet("example.parquet")

这是我使用的 CSV 文件：

col1,col2,col3
A,1,
B,2,
C,3,

我希望 col3 是镶木地板文件中的类型，而不是 INT32

Answer 1

如果列没有值，请尝试以下代码以避免错误。

import pandas as pd
df = pd.read_csv("example.csv", dtype={"col1": "str", "col2": "float", "col3": "str"}).fillna('')
df.to_parquet("example.parquet")