Q&A
やりたいこと
n×3の2次元配列が特定の列（例えば[0, 1, 3]）を含むかどうか調べたいのですが、下記のソースコードのようにfor文を使って手あたり次第調べるしかないでしょうか。
より早く調べる方法があれば教えていただけますと幸いです。
また、もしfor文を使わない方法があれば、該当の列が何番目にあるのかを調べる方法も知りたいです。
python
1a = np.random.randint(0, 10, (10000, 3)) 2for i in range(10000): 3 if(a[i][0] == 0 and a[i][1] == 1 and a[i][2] == 3): 4 print('Yes') 5 break
回答1件
0
ベストアンサー
python
1>>> import numpy as np 2>>> a = np.random.randint(0, 10, (10000, 3)) 3>>> 4>>> np.where((a==[0, 1, 3]).all(axis=1)) 5(array([2423, 3882, 6774, 7947]),) 6>>> 7>>> a[2423] 8array([0, 1, 3]) 9>>> a[3882] 10array([0, 1, 3]) 11>>> a[6774] 12array([0, 1, 3]) 13>>> a[7947] 14array([0, 1, 3])
投稿2022/10/04 10:24
編集2022/10/04 10:36
総合スコア19798
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。