Я пытаюсь использовать двумерный массив в качестве логического значения в цикле if, кажется, не могу понять это правильно, это большой расчет (400 600), поэтому я стараюсь избегать использования оператора двойного if. Вот как выглядит код.
if obstacle == 1:
f1_coll =f3
f2_coll =f4
f3_coll =f1
f4_coll =f2
else:
f1_coll=tau*(f1 +f2 -f3 +f4 )
f2_coll =tau*(f1 +f2 +f3 -f4 )
f3_coll =tau*(-f1 +f2 +f3 +f4 )
f4_coll =tau*(f1 -f2 +f3 +f4 )
все переменные имеют одинаковый размер (400 600), за исключением тау, который является константой. Есть ли муравьиная функция, которая может помочь мне с этим?