Python：如何提取数组中第一列值为1的行生成新数组

给定原始数组：

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

要筛选出第一列值为1的行，你可以通过遍历数组+条件判断的方式实现，具体方案如下：

方法一：基础循环遍历 #

初始化空数组，逐个遍历原始数组的每一行，检查每行第一个元素（索引为0）是否等于1，符合条件的行直接添加到新数组中：

new_array = []
for row in original_array:
    if row[0] == 1:
        new_array.append(row)

print(new_array)  # 输出结果: [[1,20,25],[1,45,16]]

方法二：列表推导式（简洁写法） #

如果追求代码简洁性，可以用Python的列表推导式一行完成筛选逻辑：

new_array = [row for row in original_array if row[0] == 1]
print(new_array)  # 输出结果: [[1,20,25],[1,45,16]]

两种方法最终都会得到你需要的目标数组。

内容的提问来源于stack exchange，提问作者Jack