在 pandas 中读取镶木地板文件（read_parquet()）时如何保留 dtypes？

Question

代码：

In [31]: df = pd.DataFrame({"a": [[{"b": 1}], [{"b": np.nan}]]})

In [32]: df
Out[32]:
              a
0    [{'b': 1}]
1  [{'b': nan}]

In [33]: df.dtypes
Out[33]:
a    object
dtype: object

In [34]: df.to_parquet("a.parquet")

In [35]: pd.read_parquet("a.parquet")
Out[35]:
               a
0   [{'b': 1.0}]
1  [{'b': None}]

正如您在此处看到的， [{'b': 1}]变为[{'b': 1.0}] 。

即使在阅读镶木地板文件时，我如何保持 dtypes？

Answer 1

尝试开启use_nullable_dtypes

df = pd.DataFrame({"a": [[{"b": 1}], [{"b": None}]]})
df.dtypes

df.to_parquet("a.parquet")
pd.read_parquet("a.parquet", use_nullable_dtypes=True)

	一个
0	[{'b': 1.0}]
1	[{'b'：无}]

有关更多说明，请参阅pandas 文档