要從現有數組中提取子集,可以使用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]]
通過調整切片操作的參數,可以提取想要的子集。