首页 > 解决方案 > 如何使用 pandas read_pickle 从 qrc 资源文件中读取包含 pandas 数据框的 pickle 文件?

问题描述

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

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

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

从 qrc 资源文件创建 pandas 数据框

资源文件:

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

标签: pythonpandaspyqtpyqt5qresource

解决方案


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

推荐阅读