python绘制分子能级跃迁图

要绘制分子能级跃迁图，需要使用Python中的一些库来处理和绘制数据，如numpy、matplotlib和networkx。以下是一个简单的例子，用于绘制氢分子的能级跃迁图：

import numpy as np
import matplotlib.pyplot as plt
import networkx as nx

# 构建能级跃迁矩阵
transition_matrix = np.zeros((8, 8))
transition_matrix[0,1] = 1
transition_matrix[1,2] = 1
transition_matrix[1,3] = 1
transition_matrix[2,4] = 1
transition_matrix[2,5] = 1
transition_matrix[3,6] = 1
transition_matrix[3,7] = 1

# 构建能级图
levels = [(0, 'H'), (1, 'H*'), (2, 'H**'), (3, 'H***'), (4, 'H****'), (5, 'H*****'), (6, 'H******'), (7, 'H*******')]
G = nx.DiGraph()

for level in levels:
    G.add_node(level[0], label=level[1])

for i in range(len(levels)):
    for j in range(len(levels)):
        if transition_matrix[i, j] == 1:
            G.add_edge(i, j)

# 绘制图形
pos = nx.spring_layout(G)
labels = nx.get_node_attributes(G, 'label')
nx.draw_networkx_nodes(G, pos, node_size=3000)
nx.draw_networkx_edges(G, pos, arrows=True)
nx.draw_networkx_labels(G, pos, labels)

plt.axis('off')
plt.show()

此代码将绘制以下能级跃迁图：