Python：如何在numpy数组中找到数字的值？

Question

我有以下数组：

a = np.array([[0,1,2,3,4,5,6,7],[8,9,10,11,12,13,14,15],[16,17,18,19,20,21,22,23],[24,25,26,27,28,29,30,31],[32,33,34,35,36,37,38,39,],[40,41,42,43,44,45,46,47],[48,49,50,51,52,53,54,55],[56,57,58,59,60,61,62,63]])

如何编写一个以0-63作为输入的函数，并告诉我该数字在数组中的位置？

示例：function（0）返回[0,0]; function（13）返回[5,1]

Answer 1

最短的方式（以对位顺序给出指数）是

np.argwhere(a==13)

给出：

array([[1, 5]])