次の例を考えると
d = array([[1, 2, 3],
[1, 2, 3],
[1, 3, 3],
[4, 4, 4],
[5, 5, 5]
])
最初の列に含まれるサブ配列を取得するには1
:
d[ d[:,0] == 1 ]
array([[1, 2, 3],
[1, 2, 3],
[1, 3, 3]])
1
andを含むサブ配列を (ループなしで) 取得する方法は5
? 次のようなものであってはなりません
d[ d[:,0] == [1,5] ] # ---> array([1, 2, 3])
うまくいかないのはどれですか?