如何从 numpy 数组中删除另一个数组中等于零的所有元素?

时间:2021-03-14 12:34:10

标签: python arrays numpy indices

如何从 numpy 数组中删除另一个数组中等于 0 的所有元素?
我想使用一个 numpy 数组 mask where mask==0 的索引来删除另一个相同形状的数组 array 的元素。
我试过 np.delete(array,mask==0) 之类的东西,但它给了我一个错误。

1 个答案:

答案 0 :(得分:0)

使用np.where

array = array[np.where(another_array != 0)]
相关问题