web-dev-qa-db-ja.com

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

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

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']
100
Daniel Andrén
T = [L[i] for i in Idx]
198
van

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インポートを気にするのにパフォーマンスが十分な懸念がある場合)

34
Paul

機能的なアプローチ:

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

from operator import itemgetter

print(list(itemgetter(*b)(a)))
[1, 34, 12]
8
T = map(lambda i: L[i], Idx)
7
Mehrdad Afshari

私はこれらのアプローチのどれにも満足していなかったので、整数、スライス、またはインデックスリストのいずれかによる柔軟なインデックス付けを可能にする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']
3
jedwards
L= {'a':'a','d':'d', 'h':'h'}
index= ['a','d','h'] 
for keys in index:
    print(L[keys])

Dict add目的のkeysindexに使用します

1
user4749532