NumPy數(shù)組的高級(jí)索引如何實(shí)現(xiàn)

小億
83
2024-05-06 13:26:54
欄目: 編程語言

NumPy數(shù)組的高級(jí)索引可以通過整數(shù)數(shù)組或布爾數(shù)組來實(shí)現(xiàn)。具體來說，可以使用整數(shù)數(shù)組來選擇數(shù)組中的特定元素，也可以使用布爾數(shù)組來選擇滿足特定條件的元素。

整數(shù)數(shù)組索引示例：

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])

# 使用整數(shù)數(shù)組索引
indices = np.array([0, 1, 2])
result = arr[indices]
print(result)

布爾數(shù)組索引示例：

import numpy as np

arr = np.array([[1, 2], [3, 4], [5, 6]])

# 使用布爾數(shù)組索引
mask = arr > 3
result = arr[mask]
print(result)

除了以上示例，高級(jí)索引還可以結(jié)合使用整數(shù)數(shù)組和布爾數(shù)組來選擇數(shù)組中的元素。NumPy提供了靈活的高級(jí)索引功能，可以方便地對(duì)數(shù)組進(jìn)行篩選和操作。

0 踩