当前位置:首页python > 正文

python如何取出二维数组的某一列

作者:野牛程序员:2023-07-18 06:59:25python阅读 2716

要从二维数组中提取一列,可以使用列表推导式或使用NumPy库。以下是两种方法的示例:

方法一:使用列表推导式

# 示例二维数组
array_2d = [[1, 2, 3],
            [4, 5, 6],
            [7, 8, 9]]

# 要提取的列索引
column_index = 1

# 使用列表推导式提取列
column = [row[column_index] for row in array_2d]

# 打印提取的列
print(column)

这将输出:

[2, 5, 8]

方法二:使用NumPy库

import numpy as np

# 示例二维数组
array_2d = np.array([[1, 2, 3],
                     [4, 5, 6],
                     [7, 8, 9]])

# 要提取的列索引
column_index = 1

# 使用NumPy的索引功能提取列
column = array_2d[:, column_index]

# 打印提取的列
print(column)

这将输出:

[2 5 8]

无论使用哪种方法,你都可以通过指定列索引来提取二维数组中的特定列。请确保索引值在合理范围内,不超出数组的列数。


野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击