我有一个包含 0 到 n 值的 2D Numpy 数组。我想获得一个长度为 n 的列表，以便该列表的第 i 个元素是一个包含所有值为 i+1（不包括 0）的索引的数组。
例如，对于输入
array([[1, 0, 1],
[2, 2, 0]])
我期待得到
[array([[0, 0], [0, 2]]), array([[1,0], [1,1]])]
我发现了这个相关的问题： Get a list of重复元素的所有索引在 numpy 数组 中可能会有帮助，但我希望找到一个更直接的解决方案，不需要对数组进行展平和排序，并且尽可能高效.