Python找出二维数组中某个元素索引，自定义函数