要从现有数组中提取子集,可以使用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]]
通过调整切片操作的参数,可以提取想要的子集。