Pythonでは、別のリストでリストにインデックスを付けるにはどうすればよいですか?


130

このような別のリストでリストにインデックスを付けたい

L = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
Idx = [0, 3, 7]
T = L[ Idx ]

そしてTは最終的に['a'、 'd'、 'h']を含むリストになるはずです。

より良い方法はありますか

T = []
for i in Idx:
    T.append(L[i])

print T
# Gives result ['a', 'd', 'h']

回答:


241
T = [L[i] for i in Idx]

6
これはforループよりも速いですか、それとも短いだけですか?
ダニエルアンドレ

9
@daniel:両方+推奨
SilentGhost

13
クイックタイミングテスト(pyscoまたは何もないので、それを作成してください)では、リストの理解度がループよりも2.5倍速くなっています(1000要素、10000回繰り返されます)。
ジェームズホプキン、

2
(マップとラムダを使用するとさらに遅くなります-これは、反復ごとに関数を呼び出すため、予想されます)
James Hopkin

+1索引付けリストが任意の場合、リストの圧縮が適しています。ここではそうではないようですが、可能であればスライスはさらに高速だと思います。
ハイメ、

40

numpyを使用している場合は、そのような拡張スライスを実行できます。

>>> import numpy
>>> a=numpy.array(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])
>>> Idx = [0, 3, 7]
>>> a[Idx]
array(['a', 'd', 'h'], 
      dtype='|S1')

...そしておそらくはるかに高速です(パフォーマンスがnumpyインポートに悩まされるのに十分な懸念がある場合)


5
私の簡単なtimeitテストでは、np.arrayを使用すると、実際には(配列への変換を含めて)ほぼ3倍遅いことがわかりました。
Andrzej Pronobis 16

とにかく配列操作のために変換する必要がある場合は、よりうまく機能します。通常のリスト操作には時間がかかりすぎます。
Frankliuao

9

機能的アプローチ:

a = [1,"A", 34, -123, "Hello", 12]
b = [0, 2, 5]

from operator import itemgetter

print(list(itemgetter(*b)(a)))
[1, 34, 12]

bたまたまアイテムが1つしかない場合、これは機能しません。
blhsing


5

私はこれらのアプローチのいずれにも満足していなかったのでFlexlist、整数、スライス、またはインデックスリストのいずれかによる柔軟なインデックス付けを可能にするクラスを思いつきました。

class Flexlist(list):
    def __getitem__(self, keys):
        if isinstance(keys, (int, slice)): return list.__getitem__(self, keys)
        return [self[k] for k in keys]

あなたの例では、次のように使用します。

L = Flexlist(['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h'])
Idx = [0, 3, 7]
T = L[ Idx ]

print(T)  # ['a', 'd', 'h']

これは、Pythonのパワーと柔軟性も示しています。
crowie

これを既存のコードにも拡張するのはとても簡単です。呼び出すだけでexisting_list = Flexlist(existing_list)、コードを壊すことなく必要な機能を利用できます
はい、

1
L= {'a':'a','d':'d', 'h':'h'}
index= ['a','d','h'] 
for keys in index:
    print(L[keys])

私はDict add希望keysを使用しますindex


0

また、次のように__getitem__メソッドを組み合わせて使用することもできmapます。

L = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
Idx = [0, 3, 7]
res = list(map(L.__getitem__, Idx))
print(res)
# ['a', 'd', 'h']
弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.