阅读量:0
在numpy中,我们可以使用索引来获取指定行列的数据。
例如,假设我们有一个2维的numpy数组arr
,我们想要获取第2行和第3列的数据,可以使用以下代码:
import numpy as np arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) row = 1 column = 2 # 获取指定行 row_data = arr[row, :] # 获取指定列 column_data = arr[:, column] print("第2行的数据:", row_data) print("第3列的数据:", column_data)
输出结果为:
第2行的数据: [4 5 6] 第3列的数据: [3 6 9]
我们可以使用索引arr[row, :]
来获取指定行的数据,其中row
表示行的索引。同样地,我们可以使用索引arr[:, column]
来获取指定列的数据,其中column
表示列的索引。请注意,索引从0开始计数。