溫馨提示×

numpy怎么查找元素索引

小億
251
2024-01-25 12:23:20
欄目: 編程語言

在NumPy中,可以使用np.where()函數來查找元素索引。np.where()函數返回滿足指定條件的元素的索引。以下是使用np.where()函數查找元素索引的示例:

import numpy as np

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

# 查找元素3的索引
index = np.where(arr == 3)
print(index)  # 輸出 (array([2]),)

# 查找元素大于4的索引
index = np.where(arr > 4)
print(index)  # 輸出 (array([4, 5]),)

# 查找元素不等于2的索引
index = np.where(arr != 2)
print(index)  # 輸出 (array([0, 2, 3, 4, 5]),)

注意,np.where()函數返回的是一個元組,其中包含滿足條件的元素的索引數組。如果要獲取索引數組,可以使用index[0]來獲取。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女