web-dev-qa-db-fra.com

Insérer un élément dans la liste Python après chaque nième élément

Disons que j'ai une liste Python comme celle-ci:

letters = ['a','b','c','d','e','f','g','h','i','j']

Je veux insérer un "x" après chaque nième élément, disons trois caractères dans cette liste. Le résultat devrait être:

letters = ['a','b','c','x','d','e','f','x','g','h','i','x','j']

Je comprends que je peux le faire en bouclant et en insérant. Ce que je recherche en fait, c'est une voie pythonienne, une ligne peut-être?

pythonlistindexinginsertslice
22
Hasan Iqbal

J'ai deux doublures.

Donné:

>>> letters = ['a','b','c','d','e','f','g','h','i','j']

  1. Utilisez enumerate pour obtenir l'index, ajoutez 'x' Tous les 3rd lettre, par exemple : mod(n, 3) == 2, puis concaténer en chaîne et list().

    >>> list(''.join(l + 'x' * (n % 3 == 2) for n, l in enumerate(letters)))

['a', 'b', 'c', 'x', 'd', 'e', 'f', 'x', 'g', 'h', 'i', 'x', 'j']

    Mais comme @ sancho.ssouligne cela ne fonctionne pas si l'un des éléments a plus d'une lettre.

  2. Utilisez des compréhensions imbriquées pour aplatir une liste de listes(une), découpé en groupes de 3 avec 'x' ajouté si moins de 3 à la fin de la liste.

    >>> [x for y in (letters[i:i+3] + ['x'] * (i < len(letters) - 2) for
     i in xrange(0, len(letters), 3)) for x in y]

['a', 'b', 'c', 'x', 'd', 'e', 'f', 'x', 'g', 'h', 'i', 'x', 'j']

(a) [item for subgroup in groups for item in subgroup] aplatit une liste irrégulière de listes.

13
Mark Mikofski

Essaye ça

i = n
while i < len(letters):
    letters.insert(i, 'x')
    i += (n+1)

où n correspond au nombre d'éléments que vous souhaitez insérer 'x'.

Cela fonctionne en initialisant une variable i et en la définissant sur n. Vous configurez ensuite une boucle while qui s'exécute alors que i est inférieure à la longueur de letters. Vous insérez ensuite 'x' à l'index i dans letters. Ensuite, vous devez ajouter la valeur de n+1 à i. La raison pour laquelle vous devez faire n+1 au lieu de simplement n parce que lorsque vous insérez un élément dans letters, il augmente la longueur de la liste d'une unité.

Essayer ceci avec votre exemple où n vaut 3 et vous voulez insérer 'x', cela ressemblerait à ceci

letters = ['a','b','c','d','e','f','g','h','i','j']
i = 3
while i < len(letters):
    letters.insert(i, 'x')
    i += 4

print letters

qui imprimerait

['a', 'b', 'c', 'x', 'd', 'e', 'f', 'x', 'g', 'h', 'i', 'x', 'j']

quel est votre résultat attendu.

8
michaelpri

Bien que l'utilisation de list.insert() dans une boucle for semble être plus efficace en mémoire, afin de le faire sur une seule ligne, vous pouvez également ajouter la valeur donnée à la fin de chaque bloc également divisé. divisé sur chaque nth index de la liste.

>>> from itertools import chain

>>> n = 2
>>> ele = 'x'
>>> lst = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

>>> list(chain(*[lst[i:i+n] + [ele] if len(lst[i:i+n]) == n else lst[i:i+n] for i in xrange(0, len(lst), n)]))
[0, 1, 'x', 2, 3, 'x', 4, 5, 'x', 6, 7, 'x', 8, 9, 'x', 10]
4
ozgur

Une méthode assez simple:

>>> letters = ['a','b','c','d','e','f','g','h','i','j']
>>> new_list = []
>>> n = 3
>>> for start_index in range(0, len(letters), n):
...     new_list.extend(letters[start_index:start_index+n])
...     new_list.append('x')
... 
>>> new_list.pop()
'x'
>>> new_list
['a', 'b', 'c', 'x', 'd', 'e', 'f', 'x', 'g', 'h', 'i', 'x', 'j']

Vous pouvez également utiliser la recette grouper de la documentation itertools pour la segmentation.

2
Mike Graham

C'est un vieux sujet, mais il manque la solution la plus simple et la plus "Pythonique", imo. Ce n'est qu'une extension de la partie 2 de Mark Mikofskiréponse acceptée qui améliore sans doute la lisibilité (et donc la rend plus Pythonique).

>>> letters = ['a','b','c','d','e','f','g','h','i','j']
>>> [el for y in [[el, 'x'] if idx % 3 == 2 else el for 
     idx, el in enumerate(letters)] for el in y]

['a', 'b', 'c', 'x', 'd', 'e', 'f', 'x', 'g', 'h', 'i', 'x', 'j']
1
John Paper

Je veux ajouter un nouvel élément par article.

Que dis-tu de ça ?

a=[2,4,6]
for b in range (0,len(a)):
    a.insert(b*2,1)

un est maintenant

[1, 2, 1, 4, 1, 6]
1
Tim Seed

Alors que Mark Mikofskianswer fonctionne, il existe une solution plus rapide en attribuant des tranches:

import string

# The longer the list the more speed up for list3
# letters = ['a','b','c','d','e','f','g','h','i','j']
letters = list(string.ascii_letters)
print("org:", letters)

# Use enumerate to get index, add 'x' every 3rd letter, eg: mod(n, 3) == 2, then concatenate into string and list() it.
list1 = list(''.join(l + 'x' * (n % 3 == 2) for n, l in enumerate(letters)))
print("list1:", list1)
%timeit list(''.join(l + 'x' * (n % 3 == 2) for n, l in enumerate(letters)))

# But as @sancho.s points out this doesn't work if any of the elements have more than one letter.
# Use nested comprehensions to flatten a list of lists(a), sliced in groups of 3 with 'x' added if less than 3 from end of list.
list2 = [x for y in (letters[i:i+3] + ['x'] * (i < len(letters) - 2) for i in range(0, len(letters), 3)) for x in y]
print("list2:", list2)
%timeit [x for y in (letters[i:i+3] + ['x'] * (i < len(letters) - 2) for i in range(0, len(letters), 3)) for x in y]

# Use list slice assignments
len_letters = len(letters)
len_plus_x = ll // 3
list3 = [None for _ in range(len_letters + len_plus_x)]
list3[::4] = letters[::3]
list3[2::4] = letters[2::3]
list3[1::4] = letters[1::3]
list3[3::4] = ['x' for _ in range(len_plus_x)]
print("list3:", list3)
%timeit ll = len(letters); lp = ll//3; new_letters = [None for _ in range(ll + lp)]; new_letters[::4] = letters[::3]; new_letters[2::4] = letters[2::3]; new_letters[1::4] = letters[1::3]; new_letters[3::4] = ['x' for _ in range(lp)]

produit (en utilisant le cahier jupyter)

org: ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']
list1: ['a', 'b', 'c', 'x', 'd', 'e', 'f', 'x', 'g', 'h', 'i', 'x', 'j', 'k', 'l', 'x', 'm', 'n', 'o', 'x', 'p', 'q', 'r', 'x', 's', 't', 'u', 'x', 'v', 'w', 'x', 'x', 'y', 'z', 'A', 'x', 'B', 'C', 'D', 'x', 'E', 'F', 'G', 'x', 'H', 'I', 'J', 'x', 'K', 'L', 'M', 'x', 'N', 'O', 'P', 'x', 'Q', 'R', 'S', 'x', 'T', 'U', 'V', 'x', 'W', 'X', 'Y', 'x', 'Z']
13 µs ± 1.09 µs per loop (mean ± std. dev. of 7 runs, 100000 loops each)
list2: ['a', 'b', 'c', 'x', 'd', 'e', 'f', 'x', 'g', 'h', 'i', 'x', 'j', 'k', 'l', 'x', 'm', 'n', 'o', 'x', 'p', 'q', 'r', 'x', 's', 't', 'u', 'x', 'v', 'w', 'x', 'x', 'y', 'z', 'A', 'x', 'B', 'C', 'D', 'x', 'E', 'F', 'G', 'x', 'H', 'I', 'J', 'x', 'K', 'L', 'M', 'x', 'N', 'O', 'P', 'x', 'Q', 'R', 'S', 'x', 'T', 'U', 'V', 'x', 'W', 'X', 'Y', 'x', 'Z']
13.7 µs ± 336 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)
list3: ['a', 'b', 'c', 'x', 'd', 'e', 'f', 'x', 'g', 'h', 'i', 'x', 'j', 'k', 'l', 'x', 'm', 'n', 'o', 'x', 'p', 'q', 'r', 'x', 's', 't', 'u', 'x', 'v', 'w', 'x', 'x', 'y', 'z', 'A', 'x', 'B', 'C', 'D', 'x', 'E', 'F', 'G', 'x', 'H', 'I', 'J', 'x', 'K', 'L', 'M', 'x', 'N', 'O', 'P', 'x', 'Q', 'R', 'S', 'x', 'T', 'U', 'V', 'x', 'W', 'X', 'Y', 'x', 'Z']
4.86 µs ± 35.6 ns per loop (mean ± std. dev. of 7 runs, 100000 loops each)
0
ChrisFreeman