如何修复这个索引错误？

Question

该代码接收一个 import_file ，其中包含元素周期表的元素，写入时不带空格，例如

CrCoRaLiBhMnMdSmNhPbCaUUMo...

然后程序将俄语元素周期表的元素从 json 文件读取到字典中，并将它们写入俄语的 output_file 中，也不带空格，因此该行

CrCoRa...

转变为

ХромКобальтРадий...

等等。

问题是臭名昭著的 IndexError，它发生在 p_t: 行的 if (i[n] + i[n+1]) 中。我尝试将 while n <= len(i): 更改为 while n <= len(i) - 1:，但没有帮助。

import json

def periodic_table(import_file, output_file):
    p_t = json.load(open('periodic_table.json', encoding = 'utf-8'))
    with open(import_file) as i_f, open(output_file, 'w') as o_f:
        for i in i_f.readlines():
            n = 0
            while n <= len(i):
                if (i[n] + i[n+1]) in p_t: 
                    o_f.write(p_t[i[n] + i[n+1]])
                    n += 2
                elif (i[n] + i[n+1] + i[n+2]) == 'Uue':
                    o_f.write(p_t['Uue'])
                    n += 3
                elif i[n] in p_t:
                    o_f.write(p_t[i[n]])
                    n += 1

periodic_table('import_file_3.txt', 'output_file.txt')

Answer 1

是的，你检查 i[n+1] 但你不知道它是否存在，所以你有一个 IndexError，在 python 中切片字符串也会切片，即使没有你想要切片的字符数量，所以没有索引错误。
你可以尝试这个代码：

    def periodic_table(import_file, output_file):
    p_t = json.load(open('periodic_table.json', encoding = 'utf-8'))
    with open(import_file) as i_f, open(output_file, 'w') as o_f:
        for i in i_f.readlines():
            while len(i) > 0:
                if len(i[0:2]) == 2 and i[0:2] in p_t:
                    o_f.write(i[0:2])
                    i = i[2:]
                elif i[0:3] == 'Uue':
                    o_f.write(p_t['Uue'])
                    i = i[3:]
                elif i[0] in p_t:
                    o_f.write(p_t[i[0]])
                    i = i[1:]
                else:
                    break # otherwise it will run forever if you have an error in the file

Answer 2

忽略语言翻译问题，我建议有一个离散函数，它返回元素符号列表，然后从那里开始。

def getElementList(filename):
    elist = []
    with open(filename) as elements:
        for e in map(str.strip, elements.readlines()):
            i = 0
            while i < len(e):
                e1, *e2 = e[i:i+2]
                if e2 and e2[0].islower():
                    elist.append(e1+e2[0])
                    i += 1
                else:
                    elist.append(e1)
                i += 1
    return elist

这允许将符号拆分为多行，但本身拆分为 2 行的 2 字符符号将不起作用