Numpy在给定条件的情况下对矩阵进行高级索引

Question

如何使用np索引选择具有特定条件的矩阵行？

我的矩阵是

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

我想选择第一个元素大于一的行。类似于：

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

Answer 1

编辑：form = ContactForm() form.fields['message'].placeholder = "This is a dynamic placeholder" 是可选的，感谢@ user2357112。

np.where

  

尝试

n[n[:, 0] > 1]

     

其中n[np.where(n[:, 0] > 1)] 返回满足的索引数组   给定条件。

Answer 2

您可以使用：

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