我有一个数组
arr = [[[1, 2], [3, 4]], [5, 6]]
我正在寻找一个函数 f，它返回嵌套数组中元素的索引，只给定
arr
例如
function(arr, 2) -> (0, 0, 1)
有些图书馆实现了吗？
当然如果数组是numpy数组也没问题