Python - 在列表中查找特定值的索引，该列表存储在3D numpy数组中

Question

我有以下numpy数组：

array([[[0, 0, 0],
    [0, 0, 0],
    [0, 0, 0]],

   [[0, 0, 0],
    [0, [1, 2], 0],
    [0, 0, 0]],

   [[0, 0, 0],
    [0, 3, 0],
    [0, 0, 0]]], dtype=object)

整数值是指向坐标的向量。如果坐标接近，我必须在矩阵的给定位置存储多个向量。这就是为什么我的数组中位置i = 1，j = 1和k = 1的列表[1,2]的原因。

我能够使用以下方法找到向量3的索引：

np.where(a == 3)
Out[67]: (array([2], dtype=int64), array([1], dtype=int64), array([1], dtype=int64))

但如果我对矢量1或2做同样的事情，我就不会有任何结果：

np.where(a == 1)
Out[68]: (array([], dtype=int64), array([], dtype=int64), array([], dtype=int64))

np.where(a == 2)
Out[69]: (array([], dtype=int64), array([], dtype=int64), array([], dtype=int64))

如何提取该信息？

谢谢！