AttributeError：__enter__与read_parquet一起使用with语句时

Question

我想打开特定文件夹中存在的所有镶木地板文件，并使用它绘制散点图。 我使用以下代码：

for file in glob.glob("*.parquet"):
    with pd.read_parquet(file, columns=["cordx", "cordy"]) as df:
        make some scaterplot

我可以让python成功打印文件，因此可以找到file 。 但是我AttributeError: __enter__ 。 而read_parquet是打开这些文件的正确方法。 当我不使用column = []时，我得到pyarrow.lib.ArrowIOError: Arrow error: Out of memory: malloc of size 9771487328 failed因为文件很大。 因此仅加载那些列是必要的。

Answer 1

该错误告诉您pandas.read_parquet没有实现上下文管理器，这意味着您无法使用with调用with因为__enter__和__exit__方法。 在这里查看如何将Parquet文件读入Pandas DataFrame？