如何查找numpy数组的索引并在列表中打印相应的行

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

index = np.argwhere(arr == 5)
print(index)

for i in index:
    row = arr[i[0]]
    print(row)

import numpy as np

arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [7, 8, 9]])

index = np.argwhere(arr == 5)
print(index)

for i in index:
    row = arr[i[0]]
    print(row)