使用Matplotlib画元素周期表#

进行材料计算或者查询数据库时，我们有时候想了解或者想向别人展示计算或者查询的元素的分布状况，这个时候，我们可以使用元素周期表来直观地表示元素的分布状况。 pymatviz库内置了元素周期表功能，但它的文档写得很模糊，要看懂如何使用它的元素周期表功能，需要花费一段时间，反而不如直接使用Matplotlib来画元素周期表来的简单。 我们可以使用如下代码来画元素周期表：

1. 导入库并定义全局参数和函数#

1
import matplotlib.pyplot as plt
2
from pltsci import whole_plot_set, half_plot_set, set_ticks, cm

2. 定义元素周期表#

定义元素周期表的方式其实很简单，那就是使用一个二维数组来表示元素周期表。有字的位置是元素，没有字的位置是空格。然后可以使用plt.text()函数来在指定位置添加元素的名称。

1
# 元素符号二维数组，描述周期表布局
2
element_symbols = [
3
    ['H', '', '', '', '', '', '', '', '', '', '', '', '', '', '', '', '', 'He'],
4
    ['Li', 'Be', '', '', '', '', '', '', '', '', '', '', 'B', 'C', 'N', 'O', 'F', 'Ne'],
5
    ['Na', 'Mg', '', '', '', '', '', '', '', '', '', '', 'Al', 'Si', 'P', 'S', 'Cl', 'Ar'],
6
    ['K', 'Ca', 'Sc', 'Ti', 'V', 'Cr', 'Mn', 'Fe', 'Co', 'Ni', 'Cu', 'Zn', 'Ga', 'Ge', 'As', 'Se', 'Br', 'Kr'],
7
    ['Rb', 'Sr', 'Y', 'Zr', 'Nb', 'Mo', 'Tc', 'Ru', 'Rh', 'Pd', 'Ag', 'Cd', 'In', 'Sn', 'Sb', 'Te', 'I', 'Xe'],
8
    ['Cs', 'Ba', '*', 'Hf', 'Ta', 'W', 'Re', 'Os', 'Ir', 'Pt', 'Au', 'Hg', 'Tl', 'Pb', 'Bi', 'Po', 'At', 'Rn'],
9
    ['Fr', 'Ra', '**', 'Rf', 'Db', 'Sg', 'Bh', 'Hs', 'Mt', 'Ds', 'Rg', 'Cn', 'Nh', 'Fl', 'Mc', 'Lv', 'Ts', 'Og'],
10
    ['', '', '', '', '', '', '', '', '', '', '', '', '', '', '', '', '', ''],
11
    ['', '', '*', 'La', 'Ce', 'Pr', 'Nd', 'Pm', 'Sm', 'Eu', 'Gd', 'Tb', 'Dy', 'Ho', 'Er', 'Tm', 'Yb', 'Lu'],
12
    ['', '', '**', 'Ac', 'Th', 'Pa', 'U', 'Np', 'Pu', 'Am', 'Cm', 'Bk', 'Cf', 'Es', 'Fm', 'Md', 'No', 'Lr']
13
]

1
# 创建图形和轴
2
fig, ax = plt.subplots(figsize=(20*cm, 12*cm))
3
# 添加标题
4
ax.text(0.5, 1.05, 'Periodic Table', ha='center', va='center', transform=ax.transAxes, fontsize=14, fontweight='bold')
5

6
# 绘制元素周期表
7
for i, row in enumerate(element_symbols):
8
    for j, symbol in enumerate(row):
9
        if symbol and symbol not in ['*', '**']:
10
            # 计算单元格位置
11
            cell_x = j * 1.0
12
            cell_y = -i * 1.0
13
            color = '#F5F9FD'
14
            edge_color = 'grey'
15
            fontcolor = 'dimgrey'
16
            # 绘制矩形
17
            rect = Rectangle((cell_x, cell_y), 0.95, 0.95,
18
                            facecolor=color, edgecolor=edge_color, lw=1)
19
            ax.add_patch(rect)
20
            # 添加元素符号
21
            ax.text(cell_x + 0.5, cell_y + 0.6, symbol,
22
                   ha='center', va='center', fontsize=9, fontweight='bold', color=fontcolor)
23

24
# 设置轴范围和隐藏坐标轴
25
ax.set_xlim(-0.5, 18.5) # 如果你的元素周期表显示不全，请尝试修改x、y范围
26
ax.set_ylim(-9.5, 1)
27
ax.axis('off')

结果：

使用Matplotlib画元素周期表

使用Matplotlib画元素周期表#

1. 导入库并定义全局参数和函数#

2. 定义元素周期表#

目录