PyQT :在两个TableView之间传输

from PyQt5.QtWidgets import QApplication, QMainWindow, QTableView, QPushButton, QVBoxLayout, QWidget, QAbstractItemView, QHeaderView, QMessageBox
from PyQt5.QtGui import QStandardItemModel, QStandardItem
from PyQt5.QtCore import Qt

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("Data Transfer")
        self.resize(400, 300)

        # 创建数据模型和表格控件
        self.source_model = QStandardItemModel()
        self.source_table = QTableView()
        self.source_table.setModel(self.source_model)
        self.source_table.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.source_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)

        self.destination_model = QStandardItemModel()
        self.destination_table = QTableView()
        self.destination_table.setModel(self.destination_model)
        self.destination_table.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.destination_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)

        # 创建按钮
        self.transfer_button = QPushButton("Transfer")
        self.transfer_button.clicked.connect(self.transfer_data)

        # 创建布局
        layout = QVBoxLayout()
        layout.addWidget(self.source_table)
        layout.addWidget(self.transfer_button)
        layout.addWidget(self.destination_table)

        # 创建主窗口
        central_widget = QWidget()
        central_widget.setLayout(layout)
        self.setCentralWidget(central_widget)

    def transfer_data(self):
        # 获取选中的数据
        selected_indexes = self.source_table.selectedIndexes()
        if len(selected_indexes) == 0:
            QMessageBox.warning(self, "Warning", "No data selected.")
            return

        # 移除选中的数据
        rows = set()
        for index in selected_indexes:
            rows.add(index.row())
        for row in sorted(rows, reverse=True):
            self.source_model.removeRow(row)

        # 添加到目标表格
        for index in selected_indexes:
            item = QStandardItem(index.data())
            self.destination_model.appendRow(item)

if __name__ == "__main__":
    app = QApplication([])
    window = MainWindow()
    window.show()
    app.exec_()