Использовать 2D массив в качестве условия - PullRequest
Новая
       10

Использовать 2D массив в качестве условия

0 голосов
/ 11 февраля 2020

Я пытаюсь использовать двумерный массив в качестве логического значения в цикле 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), за исключением тау, который является константой. Есть ли муравьиная функция, которая может помочь мне с этим?

...