如何基于PyQt实现Matplotlib元素周期表的交互功能

方案对比 #

• 方案1（PyQt内重写周期表）：工作量大，需重新实现布局、样式，复用现有代码的可能性极低，开发效率差。
• 方案2（整合现有Matplotlib周期表）：优先选择，已有成熟的周期表实现，只需解决Matplotlib与PyQt的整合、点击事件绑定问题，复用性高，开发成本低。

核心问题解决 #

1. 消除双窗口问题 #

当前弹出两个窗口的原因是PeriodicTablev1.plt.figure()创建的Figure被Matplotlib默认后端自动显示。解决方法：

• 若可修改PeriodicTablev1，让它提供一个返回Figure对象的函数（比如create_periodic_table()），且不调用plt.show()。
• 若无法修改PeriodicTablev1，可在导入时指定Matplotlib使用PyQt5后端，并关闭自动弹出的窗口：

  import matplotlib
  matplotlib.use('Qt5Agg')  # 强制绑定PyQt5后端
  import PeriodicTablev1
  import matplotlib.pyplot as plt
  # 获取figure后关闭默认窗口
  table_figure = PeriodicTablev1.plt.figure()
  plt.close()
  

2. 实现单元格点击交互 #

给Matplotlib Canvas绑定鼠标点击事件，判断点击位置是否落在周期表Cell上，提取元素信息后弹出PyQt窗口展示。

修改后的完整代码 #

from PyQt5.QtWidgets import *
import matplotlib
matplotlib.use('Qt5Agg')  # 强制使用PyQt5后端，避免双窗口
import PeriodicTablev1
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
import matplotlib.pyplot as plt

# 元素信息映射表，可从PeriodicTablev1中复用已有数据
ELEMENT_INFO = {
    'H': {'name': 'Hydrogen', 'group': 1, 'type': 'Nonmetal', 'valence_electrons': 1},
    'He': {'name': 'Helium', 'group': 18, 'type': 'Noble Gas', 'valence_electrons': 2},
    'Li': {'name': 'Lithium', 'group': 1, 'type': 'Alkali Metal', 'valence_electrons': 1},
    # 补充其余元素信息...
}

def main():
    app = QApplication([])
    window = QWidget()
    window.setGeometry(100, 100, 1000, 700)
    window.setWindowTitle("Interactive Periodic Table")

    layout = QVBoxLayout()

    # 获取周期表Figure并关闭Matplotlib默认窗口
    table_figure = PeriodicTablev1.plt.figure()
    plt.close()

    # 创建Canvas并添加到PyQt布局
    table_canvas = FigureCanvas(table_figure)
    layout.addWidget(table_canvas)

    # 绑定鼠标点击事件
    table_canvas.mpl_connect('button_press_event', on_cell_click)

    window.setLayout(layout)
    window.show()
    app.exec_()

def on_cell_click(event):
    # 遍历Figure内所有元素，识别周期表单元格
    for artist in event.figure.get_children():
        if isinstance(artist, plt.table.Cell):
            # 判断点击位置是否在当前Cell范围内
            if artist.get_window_extent().contains(event.x, event.y):
                element_symbol = artist.get_text().get_text()
                # 获取对应元素信息
                info = ELEMENT_INFO.get(element_symbol, {})
                # 弹出信息窗口
                msg_box = QMessageBox()
                msg_box.setWindowTitle(f"Element: {element_symbol}")
                content = (
                    f"Full Name: {info.get('name', 'Unknown')}\n"
                    f"Group: {info.get('group', 'Unknown')}\n"
                    f"Type: {info.get('type', 'Unknown')}\n"
                    f"Valence Electrons: {info.get('valence_electrons', 'Unknown')}"
                )
                msg_box.setText(content)
                msg_box.exec_()
                break

if __name__ == '__main__':
    main()

关键说明 #

1. 后端绑定：matplotlib.use('Qt5Agg')确保Matplotlib的Figure由PyQt管理，彻底避免双窗口问题。
2. 事件处理：通过mpl_connect绑定button_press_event，在回调中遍历识别Cell元素，判断点击位置是否命中单元格。
3. 信息复用：若PeriodicTablev1中已存储元素详细信息，直接替换ELEMENT_INFO即可，无需重新定义。

内容的提问来源于stack exchange，提问作者ChikN42 -