给列表中的元素编号

Question

我有一个包含元素及其权重的列表：

[['H', '1.0079'], ['He', '4.00260'], ['Li', '6.941']...]

有没有办法定义第一个对象：['H', '1.0079'] 为数字 1。然后 ['He', '4.00260'] 为数字 2 等等？目标是能够向使用该程序的用户询问随机元素的原子序数。

这是我从文件中读取有关元素的数据并将数据放入列表时的代码：

def create_list():
    atom_file = open('atomer2.txt', 'r')
    atoms = []
    for line in atom_file:
        data = line.split()
        atomer.append(data)
    atomer.sort(key=lambda x: float(x[1]))

Answer 1

如果你已经根据原子量排序，那么你可以写一个简短的一行来得到 n^日元素：

get_nth_element = lambda n: atomer[n-1]