python取出矩阵中第一列为1的行

假设矩阵是一个二维列表，可以使用列表推导式来取出第一列为1的行：

matrix = [
    [1, 2, 3],
    [1, 4, 5],
    [2, 3, 4]
]

result = [row for row in matrix if row[0] == 1]

print(result) # 输出 [[1, 2, 3], [1, 4, 5]]

在上面的代码中，`[row for row in matrix]` 表示对于矩阵中的每一行，都返回这一行的列表。`if row[0] == 1` 则表示只返回第一列为1的行。最后，将满足条件的所有行组成一个新的列表 `result` 并打印出来。