このような別のリストでリストにインデックスを付けたい
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']
このような別のリストでリストにインデックスを付けたい
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']
回答:
T = [L[i] for i in Idx]
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インポートに悩まされるのに十分な懸念がある場合)
機能的アプローチ:
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つしかない場合、これは機能しません。
私はこれらのアプローチのいずれにも満足していなかったので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']
existing_list = Flexlist(existing_list)
、コードを壊すことなく必要な機能を利用できます
L= {'a':'a','d':'d', 'h':'h'}
index= ['a','d','h']
for keys in index:
print(L[keys])
私はDict add
希望keys
を使用しますindex