溫馨提示×

NumPy索引和切片怎么使用

小億
82
2024-05-13 11:22:20
欄目: 編程語言

NumPy中的索引和切片可以幫助我們訪問和操作數(shù)組中的元素。下面是一些常見的索引和切片操作示例：

  1. 索引一個元素：
import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr[2])  # 輸出：3
  1. 切片一個數(shù)組：
import numpy as np

arr = np.array([1, 2, 3, 4, 5])
print(arr[1:4])  # 輸出：[2 3 4]
  1. 多維數(shù)組的索引和切片：
import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
print(arr[1, 2])  # 輸出：6
print(arr[:, 1])  # 輸出：[2 5 8]
print(arr[1:, :2])  # 輸出：[[4 5] [7 8]]
  1. 使用布爾索引：
import numpy as np

arr = np.array([1, 2, 3, 4, 5])
mask = arr > 2
print(arr[mask])  # 輸出：[3 4 5]

這些是一些常見的NumPy索引和切片操作示例，希望能幫助到你。

0 踩