示例问题:
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)