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

How can I read pickle file containing pandas data frame from qrc resource file with pandas read_pickle?

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

我正在使用 pickle 数据结构来存储以前创建的 DataFrame。在我的应用程序中,我正在使用 pandas 读取保存的泡菜。当我尝试从 qrc_resources(使用 pyrcc5 创建)Python 模块执行此操作时,出现错误。

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

资源文件:

<!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 文件格式,我得到了一个错误。我对数据序列化不是很熟悉,我无法弄清楚我做错了什么。

您必须使用 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)