溫馨提示×

NumPy中怎么創(chuàng)建子數(shù)組

小億
94
2024-05-11 18:22:50
欄目: 編程語言

要在NumPy中創(chuàng)建子數(shù)組，可以使用切片操作符:來選擇數(shù)組的子集。以下是一些示例代碼：

  1. 創(chuàng)建一個一維數(shù)組并選擇子數(shù)組：
import numpy as np

arr = np.array([1, 2, 3, 4, 5])
sub_arr = arr[1:4] # 選擇索引為1到3的子數(shù)組
print(sub_arr)
  1. 創(chuàng)建一個二維數(shù)組并選擇子數(shù)組：
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
sub_arr = arr[:2, 1:] # 選擇前兩行和第二列之后的子數(shù)組
print(sub_arr)
  1. 使用布爾索引選擇子數(shù)組：
arr = np.array([1, 2, 3, 4, 5])
mask = np.array([True, False, True, False, True])
sub_arr = arr[mask] # 選擇mask中為True的元素組成的子數(shù)組
print(sub_arr)

通過這些示例，您可以根據(jù)需要創(chuàng)建和選擇NumPy數(shù)組的子數(shù)組。

0 踩