python - 从列表中生成多个条件语句

Question

我有一个这样的列表

a = [1,0,6,9,6,0,6,0,1,4,0,7,5,0]

我想创建一个这样的条件语句:

if a[1,3,8,13] != 0:
    do something

而这段代码显然是错误的。它必须是这样的:

if a[1] != 0 and a[3] != 0 and a[8] != 0 and a[13] != 0:
    do something

我想知道在我的案例中是否有更优雅的方法来使用 lambda 或循环编写多个条件语句。假设我的列表长度为 100，我需要为列表的 57 列创建条件语句。我可能不想以这种方式将它们全部写出来...... 谢谢

Answer 1

1.

你可以使用all:

if all(a[i] != 0 for i in [1,3,8,13]): # => a[i] != 0 for every i 
    #do smth

或者与 any 相同:

if not any(a[i] == 0 for i in [1,3,8,13]): # => there is no such i that a[i] == 0
    #do smth                               # => a[i] != 0 for every i 

---

2.

或者，您可以使用一些函数式编程的东西。
例如，您可以尝试 filter:

a = [1,0,6,9,6,0,6,0,1,4,0,7,5,0]
idxs_to_check = [1,3,8,13]

f = list(filter(lambda i: a[i] != 0, idxs_to_check))
if f == idxs_to_check:
    print('Meow')

或者，一行中的类似条件:

if not list(filter(lambda i: a[i] == 0, [1,3,8,13])): 
    print('Meow')

---

3.

最后，在您的特定情况下，您可以使用包含零的列表的乘积为零这一事实:

from numpy import prod

if prod([a[i] for i in [1,3,8,13]]):
    print("Woof")