[已解决]python中求一个数组中，所有符合特定条件的元素的索引位置

随风飘扬的蓝

不使用for循环遍历数组，有没有简单快速的方法求一个数组中大于某个数，小于某个数的所有元素的索引位置。比如matlab中简单的函数find。

最佳答案

月排行榜 / 总排行榜

jerryxjr1220

2017-8-24 10:37:31

可以用pandas的DataFrame实现

1. from pandas import DataFrame as df
   
2. data = df([1,2,3,4,5,6,7,8,9])
   
3. print(data[data>5].dropna(how='any'))

复制代码

5  6.0
6  7.0
7  8.0
8  9.0

前面的就是索引位置，后面的是数值

跳转到最佳答案楼层

jerryxjr1220

可以用pandas的DataFrame实现

1. from pandas import DataFrame as df
   
2. data = df([1,2,3,4,5,6,7,8,9])
   
3. print(data[data>5].dropna(how='any'))

复制代码

5  6.0
6  7.0
7  8.0
8  9.0

前面的就是索引位置，后面的是数值

gunjang

不用这么麻烦，直接用列表式

1. data = [1,2,3,4,5,6,7,8,9]
   
2. print([i for i in range(len(data)) if data[i] > 5 and data[i] < 9])

复制代码

jerryxjr1220

gunjang 发表于 2017-8-24 10:55
不用这么麻烦，直接用列表式

他说不使用for循环，列表式也要调用for循环的

gunjang

jerryxjr1220 发表于 2017-8-24 11:00
他说不使用for循环，列表式也要调用for循环的

抱歉没看清，那用filter

1. data = [1,2,3,4,5,6,7,8,9]
   
2. b = list(filter(lambda x: x>5 and x<9, data))
   
3. print(b)

复制代码

这是python3, 2不用那个list

哦哦，只是返回数值，索引暂时不会