Как в Python индексировать список другим списком?

131

Я хочу проиндексировать список с другим списком, подобным этому

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']
python list indexing Даниэль Андрен
источник

Ответы:

242 
T = [L[i] for i in Idx]
фургон
источник
7
Это быстрее, чем цикл for, или только короче?
Daniel Andrén
10
@daniel: оба + рекомендуются
SilentGhost
14
Быстрый тест на время (без pysco или чего-то еще, так что делайте из этого что хотите) показал понимание списка в 2,5 раза быстрее, чем цикл (1000 элементов, повторение 10000 раз).
Джеймс Хопкин,
2
(использование карты и лямбды выполняется еще медленнее - этого и следовало ожидать, поскольку он вызывает функцию для каждой итерации)
Джеймс Хопкин,
+1 Если список индексации произвольный, то его составление подходит. Я думаю, что, когда это возможно, что здесь не так, срезы выполняются еще быстрее.
Jaime
41

Если вы используете 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 раза медленнее (включая преобразование в массив).
Анджей Пронобис
Он работает лучше, если вам все равно нужно преобразовать его для операций с массивом. Слишком много времени для обычных операций со списком.
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содержит только один элемент.
blhsing
7 
T = map(lambda i: L[i], Idx)
Мехрдад Афшари
источник
6
необходимо преобразовать в список в py3k
SilentGhost
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']
jedwards
источник
что также демонстрирует мощь и гибкость 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

user4749532
источник
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']
Дэвид С.
источник