pythonで配列間での条件付き計算

a = [-1, 2, 3] b = [-3, -2, 3]の配列があった場合、これの要素に条件をつけて計算した上、計算結果を新しい配列に入れたいです。

例えば、a[0]の要素が
a[0] <0 and b[0] >= 0 であれば、c[0] = a[0]+b[0]
a[0] >=0 and b[0] <0 であれば、c[0] = a[0] * b[0]

このように列ごと条件付きで計算して、cの配列に入れたいです。

以下のようなやり方で処理しましたが、より効率より書き方があれば、教えてください。


def makedeg(x, y):
    result = [0,0,0,0,0]

    for i in range(5):
        
     if x[i] < 0 and y[i] >= 0:
         result[i] = x[i] + y[i]
     elif x[i] >= 0 and y[i] >= 0:
         result[i] = x[i] * y[i]
     elif x[i] >= 0 and y[i] <0:
         result[i] = x[i]/y[i]
     else:
         result[i] = x[i]-y[y]

    return result

よろしくお願いします。

1T2R3M4

2022/05/13 05:36

現時点でのコードを質問に追記していただけませんか。

1T2R3M4

2022/05/13 06:58

そのエラーはどの行で出たのでしょうか。 paizaではエラー出ませんでした。 https://ddjkaamml8q8x.cloudfront.net/questions/2022-05-13/962c0f50-a0f6-464b-90bb-ce12ee1d1331.png

ParaParaDance

2022/05/13 07:05

vscodeを使ってますが、関数を実行した行がエラーになっています。

1T2R3M4

2022/05/13 07:13

>関数を実行した行がエラーこの部分のコードはどうなっているのでしょうか。

ParaParaDance

2022/05/13 07:20

すみません。function' object is not subscriptableは、解決しました。入力値に誤謬があったみたいです。そして、ソースも修正しましたが、一応、想定した通り、動いています。やはり、パイソンでも、こういったクラシックなやり方でしか方法がないのでしょうか。

行動規範の内容に同意します

回答1件

ベストアンサー

Python
1def makedeg(x, y):
2    return [a+b if a<0<=b else a*b if a>=0<=b else 
3            a/b if a>=0>b else a-b for a,b in zip(x,y)]
4
5a = [-1, 2, 3]
6b = [-3, -2, 3]
7c = makedeg(a, b)
8print(c)  # [2, -1.0, 9]

Python
1def makedeg(x, y):
2    return [(a<0<=b)*(a+b) or (a>=0<=b)*(a*b) or 
3            (a>=0>b)*(a/b) or a-b for a,b in zip(x,y)]

Python
1def makedeg(x, y):
2    calc = lambda a,b:(a<0<=b)*(a+b) or (a>=0<=b)*(a*b) or (a>=0>b)*(a/b) or a-b
3    return [calc(a,b) for a,b in zip(x,y)]