在numpy数组中查找第N个位置？

Question

给定2D静态numpy数组中的第n位，我应该如何计算该点在所述数组中的位置？例如，给定数字5表示第n位，并且数组形状为（4，2），我想接收数组中该第n位所在的位置，即位置（0，1）。数组形状为10的第五位，1对应位置（5，0），依此类推。

是否可以使用numpy函数？

Answer 1

不清楚您的意思，请尝试numpy.argwhere(...)。

来自numpy文档：

    >>> x = np.arange(6).reshape(2,3)
    >>> x
    array([[0, 1, 2],
           [3, 4, 5]])
    >>> np.argwhere(x>1)
    array([[0, 2],
           [1, 0],
           [1, 1],
           [1, 2]])