【Python】np.where(condition_array)返回已知元素值的索引

测试一：

import numpy as np

x = np.array([1, 2, 0.5, 4, 5])  # 用你实际的数组替换这里

index = np.where(x == 0.5)
 
print(f"index值                ：{index}")
print(f"index的数据类型        ：{type(index)}")
print(f"index[0]的数据类型     ：{type(index[0])}")
print(f"index[0][0]的数据类型  ：{type(index[0][0])}")


==================== RESTART: D:/Python/Python39/代码效果测试专用.py ===================
index值                ：(array([2], dtype=int64),)
index的数据类型        ：<class 'tuple'>
index[0]的数据类型     ：<class 'numpy.ndarray'>
index[0][0]的数据类型  ：<class 'numpy.int64'>

测试二：

import numpy as np

x = np.array([1, 2, 4, 5])  # 用你实际的数组替换这里

index = np.where(x == 0.5)
 
print(f"index值                ：{index}")
print(f"index的数据类型        ：{type(index)}")
print(f"index[0]的数据类型     ：{type(index[0])}")
print(f"index[0][0]的数据类型  ：{type(index[0][0])}")


==================== RESTART: D:/Python/Python39/代码效果测试专用.py ===================
index值                ：(array([], dtype=int64),)
index的数据类型        ：<class 'tuple'>
index[0]的数据类型     ：<class 'numpy.ndarray'>
Traceback (most recent call last):
  File "D:/Python/Python39/代码效果测试专用.py", line 10, in <module>
    print(f"index[0][0]的数据类型  ：{type(index[0][0])}")
IndexError: index 0 is out of bounds for axis 0 with size 0