2016-03-27 5 views
1

の最初の要素私は、xは、新しいリスト内の特定の値(x、y)の全ての数字は、そのリストをスライスするフォームリストをスライスするPythonicの方法w.r.t.タプル

x = 
[(0,1), (0,2), (0,3), ... 
(1,1), (1,3), (1,4), ... 
... 
(n,0), (n,4), ... 
] 

の組のソートされたリストを持っていると注文は保持されます。今、これは明らかにうまくいくでしょう:

y = [(a,b) for (a,b) in x if a == n] 

しかし、それは本当に遅いです。バイナリ検索でこの条件を満たす最初と最後のインデックスを見つける方が早いでしょう。 indexは値の最初のインデックスを返し、逆リストのindexは最後のインデックスを返します。 [a for (a,b) in x]を実行してリスト全体をコピーすることなく、どのように適用しますか?

+0

_ordered_リストを仮定することはできないので、 'index'はおそらくバイナリ検索をしませんが、リストのトラバーサルなので、はるかに速くなるのではないでしょうか。 –

+2

ソートされたリストを処理する方法として[bisect](https://docs.python.org/3/library/bisect.html#searching-sorted-lists)の使用を検討してください。 –

+0

開始位置と終了位置を知った後、リスト内の部分範囲(スライス)にアクセスする方法については、http://stackoverflow.com/questions/509211/explain-pythons-slice-notationまたはhttps://docs.pythonを参照してください。 .org/2/tutorial/introduction.html#strings –

答えて

2

@Liongoldによるコメントで示唆されているように、bisectを使用することができます。あなたはt[0] == 1を持つすべてのタプルtをしたいと仮定:

from bisect import bisect_left 

x = [(0, 1), (0, 2), (1, 1), (1, 2), (2, 1), (2, 2)] 

start = bisect_left(x, (1, None)) # index of the very first (1, i) 
end = bisect_left(x, (2, None)) # index after the very last (1, i) 

y = x[start:end] 

# y: [(1, 1), (1, 2)] 

あなたはbisect docsで詳細を見つけることができます。

+0

2がリストにあると仮定できない場合はどうすればよいですか? – joachim

+1

'bisect_left(x、(2、None)'は指定された要素の挿入ポイントを返します 'i'は任意の整数 'i'のため、常に最後の'(1、i ) 'これは、必要に応じて' len(x) 'にすることができます – schwobaseggl

関連する問題