python - 如何在 if 语句中使用两个条件

Question

我有以下等式：

result=[(i,j,k) for i in S for j in S for k in S if sum([i,j,k])==0]

我想在 if 语句中添加另一个条件，使我的结果集不包含 (0,0,0)。我尝试执行以下操作： result=[(i,j,k) for i in S for j in S for k in S if sum([i,j,k])==0 && (i,j,k)!=(0,0,0)]但我收到指向&&. 我测试了第一个条件的表达式，它是正确的。

Answer 1

您正在寻找and布尔运算符：

result=[(i,j,k) for i in S for j in S for k in S if sum([i,j,k])==0 and (i,j,k)!=(0,0,0)]

&&是JavaScript、Java、Perl、PHP、Ruby、Go、OCaml、Haskell、MATLAB、R、Lasso、ColdFusion、C、C# 或 C++ 布尔语法。

Answer 2

除了那个错误而不是三重嵌套for循环之外，您还可以itertools.product在此处使用来获取笛卡尔积S * S * S：

from itertools import product
result=[ x for x in product(S, repeat = 3) if sum(x)==0 and x != (0,0,0)]

演示：

>>> S = [1, -1, 0, 0]
>>> [ x for x in product(S, repeat = 3) if sum(x) == 0 and x != (0,0,0)]
[(1, -1, 0), (1, -1, 0), (1, 0, -1), (1, 0, -1), (-1, 1, 0), (-1, 1, 0), (-1, 0, 1), (-1, 0, 1), (0, 1, -1), (0, -1, 1), (0, 1, -1), (0, -1, 1)]

Answer 3

result = [(i, j, k) for i in S
          for j in S
          for k in S
          if sum([i, j, k]) == 0 and (i, j, k) != (0, 0, 0)]