I have a very large array but here I will show a simplified case:
a = np.array([[3,4,5],[8,7,6],[5,4,0]]) array([[3, 4, 5], [8, 7, 6], [5, 4, 0]]) <