在numpy张量中创建非零值数组的有效方法？

Question

我有一个3D numpy网格A[ix,iy,iz]，我通过A[ A<minval ] = 0或A[ A>maxval ] = 0等对元素进行归零过滤。然后，我想对其余的项目执行统计。现在，我要做的是：

for ai in np.reshape(A, nx*ny*nz):
    if( ai > 0 ):
        Atemp.append(ai)

然后我在Atemp上执行统计。然而，这需要相当长的时间，我想知道是否有一种更有效的方法来创建Atemp。值得注意的是，我正在处理这些数组中的几个GB数据。

注意:我不想用不同的方法过滤掉这些项目。我想将它们归零，然后在A中创建一个包含所有非零元素的临时数组。

Answer 1

另一种选择：

 Atemp = A.ravel()[np.flatnonzero(A)]