我正在尝试建立一个由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语句。
提前谢谢!
亲切的问候