要在NumPy中創建子數組,可以使用切片操作符:
來選擇數組的子集。以下是一些示例代碼:
import numpy as np
arr = np.array([1, 2, 3, 4, 5])
sub_arr = arr[1:4] # 選擇索引為1到3的子數組
print(sub_arr)
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
sub_arr = arr[:2, 1:] # 選擇前兩行和第二列之后的子數組
print(sub_arr)
arr = np.array([1, 2, 3, 4, 5])
mask = np.array([True, False, True, False, True])
sub_arr = arr[mask] # 選擇mask中為True的元素組成的子數組
print(sub_arr)
通過這些示例,您可以根據需要創建和選擇NumPy數組的子數組。