numpy怎么获取指定行列数据

2377
2024/1/3 9:47:57
栏目: 编程语言
开发者测试专用服务器限时活动,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开始计数。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: 怎么把numpy数组变成字符型