給定條件下的矩陣上的Numpy高級索引

Question

如何使用np索引來選擇具有特定條件的矩陣的行？

我矩陣是

n = np.array([[1,2],[4,5], [1,22]]) 

，我想選擇其第一個元素是大於一的行。類似於：

n[lambda x: x[0] > 1] 

Answer 1

您可以使用：

n[n[:,0] > 0, :] 

Answer 2

編輯：np.where是可選的，謝謝@ user2357112。

n[n[:, 0] > 1] 

嘗試

n[np.where(n[:, 0] > 1)] 

其中np.where返回滿足 給定條件行索引的陣列。