如何删除多维numpy数组中零值的行？

import numpy as np

# 创建一个多维NumPy数组
arr = np.array([[1, 2, 3],
                [4, 0, 6],
                [0, 0, 0],
                [7, 8, 9]])

# 使用布尔索引删除零值的行
non_zero_rows = arr[~np.all(arr == 0, axis=1)]

print(non_zero_rows)

[[1 2 3]
 [4 0 6]
 [7 8 9]]

import numpy as np

# 创建一个多维NumPy数组
arr = np.array([[1, 2, 3],
                [4, 0, 6],
                [0, 0, 0],
                [7, 8, 9]])

# 使用np.nonzero()函数获取非零元素的索引
non_zero_indices = np.nonzero(np.any(arr != 0, axis=1))

# 使用索引删除零值的行
non_zero_rows = arr[non_zero_indices]

print(non_zero_rows)

[[1 2 3]
 [4 0 6]
 [7 8 9]]