python3 より速く配列内で条件の満たす要素を抽出したい

Question

前提・実現したいこと

pythonで配列操作を行っています。適当に昇順である配列xを用意し、ランダム値より大きくなったときの要素を抽出したいです。
配列　x = [2,4,6,8,10]

ランダム値　r = 3, 理想の出力：４
r = 4, 理想の出力：6

for文を使って、出力する方法とnp.whereを使って抽出しました。それぞれ同じ処理を10000回繰り返し、プロファイリングして計算時間を測定したところ
＃1:22.152sec
＃2:0.949sec
より速く抽出する方法を教えてください。

ソースコード

import numpy as np
import random
for j in range(10000):
    x = np.arange(0,1.1,0.0001)
    r = random.random()
    x1 = x-r
#1
    for i in x:
        if r < i:
            print(i)
            break
#2
    a = np.where(x1>0)
    print(x[a[0][0]])

補足情報（FW/ツールのバージョンなど）

python3.6

Answer 1

pythonより早く、というのであれば、C言語で実装すればいかがでしょう。早いですぜ

Comments

[LouiS0616]

題意は『Pythonよりも速く』じゃなくて『Pythonで、より速く』だと思いますが...

[y_waiwai]

しつもんくん本人がどう思っているのか、をきかないとどーしよーもないですねえｗ

Answer 2

ソート済み配列の探索なのでnp.searchsortedを使ってください。使われるアルゴリズムはバイナリサーチです。

i = np.searchsorted(x, r, side="right")
result = x[i]

numpy.searchsorted — NumPy v1.15 Manual

Comments

[selva604]

np.whereより速く処理することができました。勉強になります。ありがとうございました。

Answer 3

1の速い版。

print(next(i for i in x if i > r))