如何在Python的3D数组中查找指定值的索引？

我曾尝试获取3D数组中指定值的索引，发现最简单的方法如下：

import numpy as np

# b 是一个3D数组
# 9 和 16 是我要查找索引的目标值

values = [9, 16]

for idx, x in np.ndenumerate(b):
    if x in values:
        print(f'idx: {idx}  x: {x}')

希望我的方法能帮到有需要的人 :)

内容的提问来源于stack exchange，提问作者Mateus Ribeiro