在Numpy数组中如何找到一个值的所有坐标

如果我想找到所有 3D 数组中最大值的坐标，如何找到它们？

到目前为止，这是我的代码，但它不起作用，我不明白为什么。

s = set()
elements = np.isnan(table)
numbers = table[~elements]
biggest = float(np.amax(numbers))
a = table.tolist()
for x in a:
    coordnates = np.argwhere(table == x)
    if x == biggest:
        s.add((tuple(coordinates[0]))
print(s)

例如：

table = np.array([[[ 1, 2, 3],
        [ 8, 4, 11]],

        [[ 1, 4, 4],
        [ 8, 5, 9]],

        [[ 3, 8, 6],
        [ 11, 9, 8]],

        [[ 3, 7, 6],
        [ 9, 3, 7]]])

应该返回s = {(0, 1, 2),(2, 1, 0)}

组合np.argwhere and np.max（正如 @AshwiniChaudhary 在评论中已经指出的那样）可用于查找坐标：

>>> np.argwhere(table == np.max(table))
array([[0, 1, 2],
       [2, 1, 0]], dtype=int64)

要获得集合，您可以使用集合理解（需要将子数组转换为元组，以便它们可以存储在集合中）：

>>> {tuple(coords) for coords in np.argwhere(table == np.max(table))}
{(0, 1, 2), (2, 1, 0)}