使用蒙版从numpy ndarray中删除元素

时间:2018-03-15 14:36:15

标签: python numpy

示例问题:

import numpy as np
x = np.arange(30).reshape(-1,10)
mask = np.ones((3,10),dtype=bool)
mask[[0,1,2],[3,4,5]] = False

y=x[mask]
#y.shape returns (27,)

我希望y.shape返回(3,9)。是否有一个内置函数适用于numpy ndarrays并返回相同的?我目前的解决方案只是简化数组:

y.reshape(x.shape[0],-1)

0 个答案:

没有答案