Вопрос по indexing, python, list – В Python, как мне индексировать список другим списком?

86

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

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

и T должен в конечном итоге представлять собой список, содержащий ['a', 'd' d ',' h '].

Есть ли лучший способ, чем

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

print T
# Gives result ['a', 'd', 'h']
</code>

Ваш Ответ

6   ответов
6
T = map(lambda i: L[i], Idx)
необходимо преобразовать в список в py3k
32

вы можете выполнить расширенную нарезку следующим образом:

>>> 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')

... и, вероятно, намного быстрее (если производительности достаточно, чтобы беспокоиться об импорте)

Мой быстрый тест timeit показал, что использование np.array на самом деле почти в 3 раза медленнее (включая преобразование в массив).
Это работает лучше, если вам все равно нужно конвертировать его для операций с массивами. Слишком много времени для регулярных операций со списком.
3

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!
1
L= {'a':'a','d':'d', 'h':'h'}
index= ['a','d','h'] 
for keys in index:
    print(L[keys])

Dict add желательноkeys вindex

7

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

from operator import itemgetter

print(list(itemgetter(*b)(a)))
[1, 34, 12]
179
T = [L[i] for i in Idx]
Это быстрее, чем цикл for или только короче? Daniel Andrén
@daniel: оба + рекомендуется
Быстрый тест синхронизации (без pysco или чего-либо еще, так что делайте из этого что хотите) показал, что понимание списка в 2,5 раза быстрее цикла (1000 элементов, повторяется 10000 раз).
(использование map и лямбды еще медленнее - как и следовало ожидать, так как он вызывает функцию для каждой итерации)
+1 Если список индексации произвольный, то способ составления списка - это путь. Хотя я думаю, что, когда это возможно, что, по-видимому, не так, срезы еще быстрее.

Похожие вопросы