基於行條件的numpy slice數組列

Question

給定一個2xn數組，在第一行有條件的情況下，如何將這些列切成一個新數組？

x = np.array([[1,2,3],[10,20,30]])

In:
Out[20]: 
array([[ 1,  2,  3],
       [10, 20, 30]])

說我想要第0​​行小於等於2的所有列。

In:  x[x[:]<=2]
Out: array([1, 2])

這並沒有讓我得到10和20以及1和2。我期望

array([[ 1,  2],
       [10, 20]])

放入x []以獲取切片的正確條件是什么？

Answer 1

您可以使用以下命令對第二軸進行布爾切片：

x[:,x[0]<=2]
# returns:
array([[ 1,  2],
       [10, 20]])