Объединить строки во вложенном списке на основе индекса - PullRequest
4 голосов
/ 25 февраля 2020

Я пытаюсь объединить строки, основываясь на одинаковых значениях в них (на разных индексах). На данный момент фрагмент отлично работает не более трех слов в последовательном порядке, но разбивается на четыре.

Например:

reg = [
        ['Abraham', 0.9, 1.6],
        ['King', 1.6, 2.4],
        ['Late', 2.4, 3.2],
        ['Moto', 11.3, 11.9],
        ['GP', 11.9, 12.7],
        ['Ferrari', 14.7, 15.1],
        ['GT-86', 15.1, 15.8],
        ['HP', 16.1, 16.6],
        ['Envy', 16.6, 17.0],
        ['16', 17.0, 17.4],
        ['DV', 17.4, 18.0]
    ]


temp_word = ''
result_lst = []
isBool = False

for indx, elem in enumerate(reg):
    try:
        if elem[2] == reg[indx+1][1]:
            if isBool:
                temp_word += elem[0] + reg[indx+1][0]
                result_lst.append(temp_word)
            else:
                temp_word = elem[0]
                isBool = not isBool
        else:
            temp_word = ''
    except IndexError:
        pass

print(result_lst)
#Output:

#['AbrahamKingLate', 'MotoGP', 'FerrariGT-86', 'HPEnvy', 'HPEnvyEnvy16', 'HPEnvyEnvy1616DV']   


# Desired: 
# ['AbrahamKingLate', 'MotoGP', 'FerrariGT-86', 'HPEnvy16DV']

Любая помощь будет принята с благодарностью!

Ответы [ 3 ]

1 голос
/ 25 февраля 2020

Это можно сделать довольно просто, сохранив предыдущий элемент и затем сравнив его с текущим элементом, например:

result = []
previous = None
for current in reg:
    if previous and current[1] == previous[2]:
        result[-1] += current[0]
    else:
        result.append(current[0])
    previous = current

print(result)

Вывод:

['AbrahamKingLate', 'MotoGP', 'FerrariGT-86', 'HPEnvy16DV']
0 голосов
/ 25 февраля 2020

Попробуйте это:

names = []
first_elem = reg[0][0]

for ix, l in enumerate(reg[1:]):
    if l[1] == reg[ix][-1]:
        first_elem = first_elem + l[0]
    else:
        names.append(first_elem)
        first_elem = l[0]
names.append(first_elem)
print(names)

# ['AbrahamKingLate', 'MotoGP', 'FerrariGT-86', 'HPEnvy16DV']
0 голосов
/ 25 февраля 2020

Это будет сделано:

result = [reg[0]]

for item in reg[1:]:
    r_num = result[-1][2]
    r_name = result[-1][0]
    i_num = item[1]
    i_name = item[0]
    if i_num == r_num:
        result[-1] = [r_name+i_name] + item[1:]
    else:
        result.append(item)

result = [r[0] for r in result]

result
# ['AbrahamKingLate', 'MotoGP', 'FerrariGT-86', 'HPEnvy16DV']
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...