现在我有一个浮点值的 2D Numpy 数组,即
a
,它的形状为(10^6, 3)。
我想知道哪些行大于np.array([25.0, 25.0, 25.0])
。然后输出满足此条件的行。
我的代码如下。# Create an empty array
a_cut = np.empty(shape=(0, 3), dtype=float)
minimum = np.array([25.0, 25.0, 25.0])
for i in range(len(a)):
if a[i,:].all() > minimum.all():
a_cut = np.append(a_cut, a[i,:], axis=0)
然而,这段代码效率低下。几个小时后,结果仍未出现。 那么有没有办法提高循环速度?
axis=1
,即所有的行都是 True,也就是说所有的行都大于 25 ;) - Chrisfor
循环是不太有效的。也许有一些方法可以实现我的目标。 - Stephen Wonga[(a > minimum).all(1)]
。 - Divakar