Python 更快的“如果”使用

Question

我有一个清单：

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]

有多个if检查数字的用法在列表中。

while True:
    if 3 in a:
        some_work1  #(different)
    
    if 4 in a:
        some_work2  #(different)
    
    if 8 in a:
        some_work3  #(different)
    
    if 11 in a:
        some_work4  #(different)

    if 12 in a:
        some_work5  #(different)

对于这些多重if用法，是否有更快（更少 cpu 使用）的方法？ （列表a总是相同的。它也不会随着迭代而改变。）。 列表中没有重复的项目。 作品不重叠。

Python 3.8.7

Answer 1

使用具有恒定插入和检索时间的集合。 相比之下， in运算符a每次检查中执行线性搜索。

如果没有看到更大的代码，我不确定你的用例是什么。 我假设您的用例将a视为标志列表。 因此，一套适合该法案。

a = [1, 2, 3, 4, 5, 6, 7, 8, 9 10, 11, 12]
a = set(a)  # pass an iterable

# or simply
a = {1, 2, 3, 4, 5, 6, 7, 8, 9 10, 11, 12}

# or built at runtime
a = set()
a.add(1)
a.add(2)

if 3 in a:
    some_work1 

如果你想要更高效的 switch 语句，你已经找到了。 Python 为此使用if..elif 。 这确保了每个短路都按顺序进行评估。 如果您可以匹配多个结果，请使用字典（例如{3: functor3, 4: functor4, ...} 。函子是可调用的，即它定义了__call__()方法lambda也满足这一点。

集合是不允许重复的无序集合。 它就像一本字典，但删除了值，只留下键。 如您所知，字典键是唯一的，同样，集合的成员也是唯一的。 在这里，我们只需要一组性能。

Answer 2

选项1

您可以使用一个字典，其键是a中的数字，值对应的 function。 然后你遍历它们一次并将所需的函数存储在一个数组（ to_call ）中。 在 while 循环中，您只需遍历此数组并调用其成员。

def some_work1():
   print("work1")

def some_work2():
   print("work2");

a = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]
func = {3:some_work1,4:some_work2}

to_call = []
for k in func:
   if k in a:
      to_call.append(func[k])


while 1:
   for f in to_call:
      f();

选项 2

编写某种代码生成器，读取并生成包含 function 调用a .py 文件。