python 如何获得二维数组的某一列

在Python中，你可以使用列表推导（list comprehension）来获取二维数组（其实就是列表的列表）中的某一列。以下是一个示例代码，展示了如何获取二维数组的第col_index列：

def get_column(matrix, col_index):
    return [row[col_index] for row in matrix]
 
# 示例使用
matrix = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
 
# 获取第2列
column = get_column(matrix, 1)
print(column)  # 输出: [2, 5, 8]

这段代码定义了一个get_column函数，它接受一个二维数组matrix和一个列索引col_index，然后使用列表推导来提取第col_index列的所有元素。最后，我们创建了一个示例矩阵并调用函数来获取第二列。

参考：百度AI