前提・実現したいこと
ある2つの配列の中身が両方とも0でない時、配列の要素を別の配列の要素で置き換える処理を実行するコードを書きましたが、2重ループを使っているため、処理が遅いです。なので、高速化したいのですが、うまくいきませんでした。
発生している問題・エラーメッセージ
2重ループを以下の方法で実装を試みた結果、次のようなエラーが出ました。
ar[list_1 != 0 and list_2 != 0 and list_3 != 0] = ar2
ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()
エラーについても調べてみたところ、any()、all()を使えば解決するのではというところまではたどり着きましたが、使い方がよくわかりません。
該当のソースコード
python
1import numpy as np 2 3def main(): 4 ar = np.ones((5,5)) 5 ar2 = np.zeros((5,5)) 6 7 list_1 = np.ones((5,5)) 8 list_2 = np.ones((5,5)) 9 list_3 = np.ones((5,5)) 10 height, width = list_1.shape[0:2] 11 12 y = 0 13 x = 0 14 15 #ar[list_1 != 0 and list_2 != 0 and list_3 != 0] = ar2 16 17 for y in range(height): 18 for x in range(width): 19 if list_1[y, x] != 0 and list_2[y, x] != 0 and list_3[y, x] != 0: 20 ar[y, x] = ar2[y, x] 21 22 print(ar) 23 24if __name__ == '__main__': 25 main()
回答3件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。