提高字符串fi的效率

Question

我有一个包含许多字符串的长文本文件。以下是文件的一部分：

tyh89= 13
kb2= 0
78%= yes
##@bb1= 7634.0
iih54= 121
fgddd= no
#aa1= 0
#aa2= 1
#$ac3= 0
yt#@hh= 0
#j= 12.1
##hf= no

所以，基本上所有元素都有一个共同的结构：header= value。我的目标是搜索元素，其标题包含特定的字符串部分，并读取这些元素的值。你知道吗

现在我用一种相当直接的方法来做：将整个文件作为字符串打开/读取，将其区分为元素列表，并使用for循环对所有元素运行if/elif条件。我在下面提供我的代码。你知道吗

这是最有效的方法吗？或者有没有更有效的方法不实现循环？

def main():
    print(list(import_param()))

def import_param():
    fl = open('filename','r')
    cn = fl.read()
    cn = cn.split('\n')
    fl.close()
    for st in cn:
        if 'fgddd' in st:
            el = st.split(' ')
            yield float(el[1])
        elif '#j' in st:
            el = st.split(' ')
            yield float(el[1])    

if __name__ == '__main__': main()

Answer 1

是的，有。如果字符串包含字符串，则必须避免测试，而应关注字符串相等性。你知道吗

一旦满足相等条件，就意味着您可以使用已知关键字创建一个集合，根据=进行拆分，并测试该集合是否包含您的值（使用O(1)查找）：

key_set = {"fgddd","#j"}

for st in cn:
    if '=' in st:
       key,value = st.split("=",1)
       if key in key_set:
           el = value.strip()
           yield float(el)

如果您有不同的类型，请使用字典根据键转换为正确的类型

key_set = {"fgddd":float ,"#j": float, "whatever":int , "something":str}

for st in cn:
    if '=' in st:
       key,value = st.split("=",1)
       if key in key_set:
           el = value.strip()
           yield key_set[key](el)  # apply type conversion

请注意，如果您不希望进行任何转换，str将在传递字符串时返回自身。你知道吗

最后一点注意：如果您对输入格式有发言权，建议使用json而不是自定义格式。使用json模块解析变得很简单，过滤可以通过我所展示的相同方式实现。你知道吗