Czy mogę korzystać z informacji indeksu w funkcji mapy?

Question

Załóżmy, że istnieje lista a = [1, 3, 5, 6, 8].

Chcę zastosować transformację na tej liście i chcę uniknąć robienia tego sekwencyjnie, więc coś takiego jak map(someTransformationFunction, a) normalnie by załatwiło sprawę, ale co, jeśli transformacja musi mieć wiedzę o indeksie każdego obiektu?

Na przykład powiedzmy, że każdy element musi być pomnożony przez jego pozycję. Tak więc lista powinna zostać przekształcona na a = [0, 3, 10, 18, 32].

Czy jest jakiś sposób na zrobienie tego?

Answer 1

Użyj funkcji enumerate() dodać indeksach:

map(function, enumerate(a)) 

Twoja funkcja zostanie przekazany do krotki, z (index, value). W Pythonie 2, można określić, że Python rozpakowania krotki dla Ciebie w podpisie funkcji:

map(lambda (i, el): i * el, enumerate(a)) 

Zanotować (i, el) krotka w specyfikacji argumentów lambda. Można zrobić to samo w def stwierdzeniem:

def mapfunction((i, el)): 
    return i * el 

map(mapfunction, enumerate(a)) 

aby zrobić miejsce dla innych funkcji, takich jak funkcja podpis adnotacji, krotka rozpakowaniu w argumentach funkcji została usunięta z Pythona 3.

Demo:

>>> a = [1, 3, 5, 6, 8] 
>>> def mapfunction((i, el)): 
...  return i * el 
... 
>>> map(lambda (i, el): i * el, enumerate(a)) 
[0, 3, 10, 18, 32] 
>>> map(mapfunction, enumerate(a)) 
[0, 3, 10, 18, 32] 

Answer 2

Można użyć enumerate():

a = [1, 3, 5, 6, 8] 

answer = map(lambda (idx, value): idx*value, enumerate(a)) 
print(answer) 

Wyjście

[0, 3, 10, 18, 32] 

Answer 3

Aby przedłużyć doskonałą odpowiedź Martijn Pieters', można również używać w połączeniu z list comprehensionsenumerate:

>>> a = [1, 3, 5, 6, 8] 
>>> [i * v for i, v in enumerate(a)] 
[0, 3, 10, 18, 32] 

lub

[mapfunction(i, v) for i, v in enumerate(a)] 

czuję listowych często są bardziej czytelne niż map/lambda konstruktów. Podczas używania nazwanej funkcji mapującej, która akceptuje bezpośrednio tupot (i, v), prawdopodobnie wygrywa map.