Numpy -如何选择最后一个元素为0的行？

import numpy as np

# 创建一个示例数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [0, 0, 0]])

# 选择最后一个元素为0的行
selected_rows = arr[arr[-1] == 0]

print(selected_rows)

[[0 0 0]]

import numpy as np

# 创建一个示例数组
arr = np.array([[1, 2, 3],
                [4, 5, 6],
                [0, 0, 0]])

# 选择最后一个元素为0的行
selected_rows = arr[(arr[:, -1] == 0)]

print(selected_rows)

[[0 0 0]]