是否可以在Python中具有成员资格的内联if语句?

时间:2019-05-14 15:51:46

标签: python if-statement membership

我正在尝试建立一个由9个分量组成的带状矩阵,以解决有限差分问题。我的想法是先将它们放在标准矩阵中,然后使用某种scipy.sparse方法将其对角线化。

我的代码可以在正常的条件下正常运行,例如:

for l,c in enumerate(node_n):
    Pxx[l,4] = epsi
    if (c[0]+1,c[1]) in tensor:
        Pxx[l,5] = psi 

但是我想知道是否可以像这样在一行中编写它:

Pxx[l,2] = gamma if ((c[0]+1,c[1]+1) in tensor)

我尝试过使用方括号和不使用方括号,使用方括号等等,但是我总是得到invalid syntax。我知道这不是一个大问题,但我希望有9行代码,每个组件一行,而不是9条if语句。

提前谢谢!

亲切的问候

0 个答案:

没有答案