我正在尝试将一个1D数组与一个2D数组进行逐元素比较，并返回符合条件的2D数组元素，返回的形式仍然是2D数组，并且不使用for循环。最好使用numpy或更快的方法。

a = range(1,10)
Tna = np.random.choice(a, size=[250,10,1000], replace=True)
sum_Ta = np.sum(Tna, axis = 1)
percent = np.percentile(sum_Ta, 5, axis =0)

现在我想得到一个2D数组，其中包含sum_Ta中小于percent的元素。这样，sum_Ta的250个元素将与percent的一个元素进行1000次比较。最初我可以这样做，ES = sum_Ta[sum_Ta < percent[:,None]]，但这只会给我一个1D数组，而不是一个2D数组。

回答：

假设您的意思是，对于每一行，如果该行的元素小于与其列相关的百分比，则应包含该元素。

请尝试以下方法：

mask = sum_Ta < (percent * np.ones((250,1)))
ES = np.zeros((250, 1000))
ES[mask] = sum_Ta[mask]