Python：抓取数组的子集，其中第一列的值等于特定值答案

【问题标题】：Python: grab subset of array where value of first column is equal to specific valuePython：抓取数组的子集，其中第一列的值等于特定值
【发布时间】：2022-09-28 23:48:26
【问题描述】：

我有一个像下面这样的数组

[[1,20,25],[1,45,16],[2,67,81],[3,1,1],[3,23,22]]

我想从上面的第一个数组创建一个新数组，但只取第一列值为 1 的行。如何循环遍历整个数组，检查每行的第一列是否为 1，然后添加该行到一个新数组，使其如下所示：

[[1,20,25],[1,45,16]]

ar[ar[:, 0] == 1]

标签： python arrays numpy

【解决方案1】：

l = [[1,20,25],[1,45,16],[2,67,81],[3,1,1],[3,23,22]]
l = [x for x in l if x[0] == 1]

【讨论】：