find row or column containing maximum value in numpy array,"np.argmax(np.max(x, axis=1))"