要使用numpy索引的元组

Question

它让我发疯，但我无法理解 我有一个（10000,4）

的数据矩阵

我需要选择列0的元素

的一些行

btnClick.userInteractionEnabled = true;

我只想保留这些数据

ind1=np.where( (data[:,0]>55) & (data[:,0]<65) )

但是keep_data现在是（1,10000,4）

为什么？

P.S。 我所做的就是efollwing

keep_data=data[ind1,:]

Answer 1

numpy.where返回一个元组。

因此，请使用ind1 = np.where((data[:,0]>55) & (data[:,0]<65))[0]

注意[0]索引以选​​择元组的唯一元素。

docs：

中注明了这一点

  

numpy.where （条件[，x，y] ）

     

根据条件从x或y返回元素。

     

如果仅给出条件，返回元组 condition.nonzero()，   条件为True的索引。