具有行特定条件的过滤器numpy数组

Question

假设我有一个2d numpy数组，并且我想过滤每个行通过特定条件的元素。例如，我只想要其特定行以上的第90百分位以上的元素。我想出了这个解决方案：

import numpy as np 
a = np.random.random((6,5)) 
thresholds = np.percentile(a, 90, axis=1) 
threshold_2d = np.vstack([thresholds]*a.shape[1]).T 
mask = a > threshold_2d 
final = np.where(mask, a, np.nan) 

它的工作原理和它的矢量，但感觉有点尴尬，尤其是在我创建threshold_2d一部分。有没有更优雅的方式？我能以某种方式自动广播np.where的条件而不必创建匹配的2d掩码？

Answer 1

广播

In [36]: np.random.seed(1023) 

In [37]: a = np.random.random((6,5)) 

In [38]: thresholds = np.percentile(a, 90, axis=1) 

In [39]: threshold_2d = np.vstack([thresholds]*a.shape[1]).T 

In [40]: a>threshold_2d 
Out[40]: 
array([[ True, False, False, False, False], 
     [False, False, True, False, False], 
     [False, True, False, False, False], 
     [False, False, False, True, False], 
     [False, False, False, False, True], 
     [False, True, False, False, False]], dtype=bool) 

In [41]: a>thresholds[:,np.newaxis] 
Out[41]: 
array([[ True, False, False, False, False], 
     [False, False, True, False, False], 
     [False, True, False, False, False], 
     [False, False, False, True, False], 
     [False, False, False, False, True], 
     [False, True, False, False, False]], dtype=bool) 

In [42]: 

numpy.newaxis创建长度之一的轴，所得到的阵列视图具有的尺寸（6,1），并且可以与a arrray进行广播。