python 删除多维数组里的零值
时间: 2024-05-19 14:12:23 浏览: 153
PHP递归删除多维数组中的某个值
可以使用numpy库中的函数np.nonzero(),然后再利用np.delete()删除对应的元素即可。具体实现方法可以如下:
import numpy as np
arr = np.array([[1, 0, 3], [4, 0, 6], [7, 8, 0]])
zero_index = np.nonzero(arr == 0) # 找到为0的索引值
arr = np.delete(arr, zero_index[0], axis=0) # 删除为0的行
arr = np.delete(arr, zero_index[1], axis=1) # 删除为0的列
print(arr)
阅读全文