NumPy怎么从现有数组中提取子集

numpy

1849

2024/4/28 11:30:14

栏目: 编程语言

要从现有数组中提取子集，可以使用NumPy中的切片操作。以下是一个示例：

import numpy as np

# 创建一个示例数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

# 从数组中提取子集
subset = arr[:2, 1:]

print(subset)

在上面的示例中，我们创建了一个3x3的数组arr，并使用切片操作[:2, 1:]提取了子集。这里[:2]表示提取前两行，[1:]表示提取从第二列开始的所有列。最终得到的子集是一个2x2的数组：

[[2 3]
 [5 6]]

通过调整切片操作的参数，可以提取想要的子集。

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

NumPy怎么从现有数组中提取子集

最新知识库

相关标签