如何使用 pandas read_pickle 从 qrc 资源文件中读取包含 pandas 数据框的 pickle 文件？

Question

我在 PyQt5 中创建了简单的 UI 应用程序。我想将我所有的资源文件都放在 qrc 资源中。

我正在使用 pickle 数据结构来存储之前创建的 DataFrame。在我的应用程序中，我正在用熊猫阅读保存的泡菜。当我尝试从 qrc_resources（使用 pyrcc5 创建）Python 模块执行此操作时，出现错误。

我使用了与此答案相同的方法：

Create a pandas dataframe from a qrc resource file

资源文件：

<!DOCTYPE RCC><RCC version="1.0">
<qresource>
    <file alias="AA_data.pkl">resources/AA_data.pkl</file>
</qresource>
</RCC>

Python 代码：

import bisect, io
import pandas as pd

from PyQt5.QtGui import QImage
from PyQt5.QtCore import QFile, QIODevice
import qrc_resources       

file = QFile(':/AA_data.pkl')
if file.open(QIODevice.ReadOnly):
    f = io.BytesIO(file.readAll().data())
    AA_df = pd.read_pickle(f)

---

Error:
ValueError: Unrecognized compression type: infer

如果我对 Excel 文件进行类似操作，它就可以工作。但是使用pickle文件格式我得到一个错误。我对数据序列化不是很熟悉，我无法弄清楚我做错了什么。

Answer 1

您必须使用 None 进行压缩：

import io
import pandas as pd
from PyQt5.QtCore import QFile, QIODevice
import qrc_resources

file = QFile(':/AA_data.pkl')
if file.open(QIODevice.ReadOnly):
    f = io.BytesIO(file.readAll().data())
    AA_df = pd.read_pickle(f, compression=None)
    print(AA_df)