Numpy矩阵怎么等间隔抽取数据

794
2023/12/31 9:11:57
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

可以使用numpy的arange函数来实现等间隔抽取数据。arange函数可以生成一个等间隔的数组,可以指定起始值、终止值和步长。

下面是一个示例代码,演示如何使用arange函数来等间隔抽取数据:

import numpy as np

# 创建一个矩阵
matrix = np.array([[1, 2, 3],
                   [4, 5, 6],
                   [7, 8, 9]])

# 按行等间隔抽取数据
row_indices = np.arange(0, matrix.shape[0], 2)
row_samples = matrix[row_indices, :]
print("按行等间隔抽取数据:")
print(row_samples)

# 按列等间隔抽取数据
col_indices = np.arange(0, matrix.shape[1], 2)
col_samples = matrix[:, col_indices]
print("按列等间隔抽取数据:")
print(col_samples)

输出结果为:

按行等间隔抽取数据:
[[1 2 3]
 [7 8 9]]
按列等间隔抽取数据:
[[1 3]
 [4 6]
 [7 9]]

在上面的代码中,我们先使用arange函数生成行或列的索引,然后使用这些索引来抽取数据。其中,arange函数的第一个参数是起始值,第二个参数是终止值(不包含),第三个参数是步长。

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

推荐阅读: NumPy数组与SciPy库怎么结合使用