博客

在Python中，可以通过下标或切片来访问数组的部分元素，但如果想要检索整行或整列，需要用到切片和索引的组合。

以下是在Python中检索数组整行或整列的完整攻略：

检索整行

可以通过以下方式检索整行：

import numpy as np

# 创建3x3的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 检索第2行，注意这里的索引从0开始
row2 = arr[1]

print(row2)

输出结果为：

[4 5 6]

检索整列

可以通过以下方式检索整列：

import numpy as np

# 创建3x3的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 检索第2列，注意这里的索引从0开始
col2 = arr[:, 1]

print(col2)

输出结果为：

[2 5 8]

检索多行或多列

如果需要检索多行或多列，可以使用切片来实现，以下是一个检索二维数组中第1和第3列，以及第2和第3行的示例：

import numpy as np

# 创建3x3的二维数组
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 检索第1和第3列，以及第2和第3行
sub_arr = arr[1:, :2]

print(sub_arr)

输出结果为：

[[4 5]
 [7 8]]

以上就是在Python中检索数组整行或整列的完整攻略。要点就是要理解数组的索引和切片特性，并且根据需要组合使用。