Как сделать функцию, которая работала бы с любыми списками?

Question

[13_nastya_13]

Есть список с элементами(некоторые из них повторяются). Нужно создать другой список, с нумерациями этих слов. Как сделать функцию, которая работала бы с любыми списками?

Например для списка ['abc', 'apple', 'pen', 'abc', 'pen', 'pc'] список с номерами должен быть [0, 1, 2, 0, 2, 3]

Comments

[dmshar]

А почему 'pc' соответствует номеру "4". И куда делся номер "3"?
Почему 'pen' соответствует номеру "2", хотя а алфавитном порядке 'pc' должен идти перед 'pen'. В общем, как-то нелогично вы построили пример..

[13_nastya_13]

dmshar, опечатка, должно быть 3 вместо 4

Answer 1

[Andrey Dugin]

items = ['abc', 'apple', 'pen', 'abc', 'pen', 'pc']

Вариант #1:

cache = {}
[cache.setdefault(item, len(cache)) for item in items]

Вариант #2:

from functools import lru_cache
from itertools import count

@lru_cache(maxsize=None)
def enumerate_as_seen(_, counter=count()):
    return next(counter)

list(map(enumerate_as_seen, items))

Answer 2

[dmshar]

Если следовать обшепринятой логике - номер строиться исходя из лексикографического порядка исходного массива, то все решается в одну строчку:

lt= ['abc', 'apple', 'pen', 'abc', 'pen', 'pc']
lt_num=[sorted(list(set(lt))).index(x) for x in lt]

Результат:
[0, 1, 3, 0, 3, 2]

Comments

[13_nastya_13]

Результат должен быть [0, 1, 2, 0, 2, 3]

[Andrey Dugin]

Слишком наворочено)

[dmshar]

Andrey Dugin, специально для любителей однострочных решений :-). Я тоже считаю, что читабальность и понимаемость - столь же важны, как правильность. Но если это кому-то для олимпиады - то может и сгодиться.

Answer 3

[Влад Григорьев]

>>> l = ['abc', 'apple', 'pen', 'abc', 'pen', 'pc']
>>> index_els = {v: k for k, v in reversed(tuple(enumerate(l)))}
>>> result = [index_els[k] for k in l]
>>> result
[0, 1, 2, 0, 2, 5]

Comments

[13_nastya_13]

С другим списком не работает (

[Влад Григорьев]

13_nastya_13, будет работать с любым плоским списком в котором нет не хешируемых типов

[13_nastya_13]

Влад Григорьев, должно быть 3 вместо 5

[Влад Григорьев]

13_nastya_13, ну тогда все еще проще, и надо чуть конкретнее формулировать вопрос

def unique(old_els):
    els = []
    for el in old_els:
        if el not in els:
            els.append(el)
    return els


l = ['abc', 'apple', 'pen', 'abc', 'pen', 'pc']
indexes = {el: i for i, el in enumerate(unique(l))}
result = [indexes[k] for k in l]
print(indexes, result)

[13_nastya_13]

Влад Григорьев, простите, я первый раз задавала вопрос. Но мне решение слегка не понятно и можно ли это все написать в одной функции? (Дело в том что мне нужно эту функцию вставить в программу, которая будет проходить несколько тестов и какие списки там будут, я не знаю)

[Влад Григорьев]

13_nastya_13, можно

import random
import string


def get_indexes(els):
    els_unique = []
    for el in els:
        if el not in els_unique:
            els_unique.append(el)
    indexes = {el: i for i, el in enumerate(els_unique)}
    return [indexes[k] for k in els]


l = ['abc', 'apple', 'pen', 'abc', 'pen', 'pc']
new_l = random.sample(tuple(random.randint(100, 110) for _ in range(30)), 15)
random.shuffle(new_l)
print(get_indexes(l))
print(new_l, get_indexes(new_l))

Answer 4

[Имя Фамилия]

Используйте словарь.

Пример: lis = {'1':'abc', '2':'apple'}

Comments

[13_nastya_13]

Дело в том что я не знаю какие могут быть слова в списке, поэтому не знаю как можно использовать в моем случае словарь

Answer 5

[dooMoob]

Значит вам нужен промежуточный словарь, где ключом будет элемент из списка, а значением его индекс

Comments

[13_nastya_13]

Я думала об этом, но не смогла реализовать