python 获取多维数组的某一列
在Python中，你可以使用列表推导（list comprehension）来获取多维数组（在Python中通常是指列表的列表，或者说是一个嵌套列表）中的某一列。以下是一个简单的例子：
# 假设有一个二维数组（列表的列表）
matrix = [
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
]
# 获取第二列的元素
column = [row[1] for row in matrix]
print(column) # 输出将会是 [2, 5, 8]如果你是指的是NumPy数组，那么可以使用NumPy的索引功能：
import numpy as np
# 创建一个NumPy数组
array = np.array([
[1, 2, 3],
[4, 5, 6],
[7, 8, 9]
])
# 获取第二列的元素
column = array[:, 1]
print(column) # 输出将会是 array([2, 5, 8])在这两个例子中，我们都假设了你想获取的是第二列的元素。第一个例子适用于普通的列表的列表，第二个例子适用于使用NumPy库创建的数组。
评论已关闭