如何从QThreadPool Worker中获取pandas DataFrame？

核心问题分析 #

你的代码存在两个关键错误：

• 类方法缩进错误：get_data、return_data1、return_data2被错误定义在__init__方法内部，导致它们不是MainWindow的成员方法，执行时信号槽绑定异常，且connect返回的连接对象被意外打印。
• 异步时序问题：线程池任务是异步执行的，启动线程后立即打印self.data1和self.data2时，数据还未被赋值，此时它们仍是初始的None。

修正步骤 #

1. 调整方法缩进：将get_data、return_data1、return_data2改为MainWindow的成员方法，与__init__同级缩进。
2. 处理异步完成逻辑：通过监听finished信号统计任务完成数量，当两个任务都完成后再打印或使用获取到的DataFrame。

完整修正代码 #

import sys
import traceback
import pandas as pd
from PySide6 import QtWidgets, QtCore
from PySide6.QtCore import QRunnable, Slot, QObject, Signal

def get_data_1():
    # 模拟接口请求返回DataFrame
    return pd.DataFrame({'value': [5,10,25,53], 'price': [2.24, 2.34, 5.22, 8.66]})

def get_data_2():
    # 模拟接口请求返回DataFrame
    return pd.DataFrame({'item_id': [10,20,75,103], 'price': [10.00, 5.29, 2.33, 8.77]})

class WorkerSignals(QObject):
    finished = Signal()  
    error = Signal(tuple)
    result = Signal(object)

class Worker(QRunnable):
    def __init__(self, fn, *args, **kwargs):
        super(Worker, self).__init__()
        self.fn = fn
        self.args = args
        self.kwargs = kwargs
        self.signals = WorkerSignals()

    @Slot()
    def run(self):
        try:
            result = self.fn(*self.args, **self.kwargs)
        except:
            traceback.print_exc()
            exctype, value = sys.exc_info()[:2]
            self.signals.error.emit((exctype, value, traceback.format_exc()))
        else:
            self.signals.result.emit(result)
        finally:
            self.signals.finished.emit()

class MainWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()
        self.data1 = None
        self.data2 = None
        self.completed_tasks = 0
        self.threadpool = QtCore.QThreadPool()
        print("Multithreading with maximum %d threads" % self.threadpool.maxThreadCount())

        self.get_data()

    def get_data(self):
        # 启动第一个任务
        worker1 = Worker(get_data_1)
        worker1.signals.result.connect(self.return_data1)
        worker1.signals.finished.connect(self.check_all_completed)
        self.threadpool.start(worker1)

        # 启动第二个任务
        worker2 = Worker(get_data_2)
        worker2.signals.result.connect(self.return_data2)
        worker2.signals.finished.connect(self.check_all_completed)
        self.threadpool.start(worker2)

    def return_data1(self, df):
        self.data1 = df
        print("Data1获取完成")

    def return_data2(self, df):
        self.data2 = df
        print("Data2获取完成")

    def check_all_completed(self):
        self.completed_tasks += 1
        if self.completed_tasks == 2:
            print("\n所有数据获取完成：")
            print("Data1:\n", self.data1)
            print("\nData2:\n", self.data2)
            # 这里可以添加后续业务处理逻辑

if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec())

代码说明 #

• 缩进修正：所有业务方法作为类成员存在，确保信号槽正确绑定。
• 任务完成监测：通过completed_tasks计数器，在两个任务都发出finished信号后再打印数据，保证数据已被正确赋值。
• 信号槽绑定：每个Worker的finished信号连接到统一的完成检查方法，避免重复逻辑。

内容的提问来源于stack exchange，提问作者David Frick