web-dev-qa-db-ja.com

n番目の要素ごとにPythonリストに要素を挿入

Pythonこのようなリストがあるとします：

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

N番目の要素ごとに「x」を挿入したいと思います。そのリストの3文字としましょう。結果は次のようになります。

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

ループと挿入でそれができることを理解しています。私が実際に探しているのは、パイソン風、おそらくはワンライナーですか？

pythonlistindexinginsertslice
22
Hasan Iqbal

私は2つのライナーを持っています。

与えられた：

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

  1. enumerateを使用してインデックスを取得し、3ごとに_'x'_を追加しますrd letter、eg：mod(n, 3) == 2、次に文字列とlist() itに連結します。

    _>>> 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']
_

    しかし、 @ sancho.spoints out のように、要素のいずれかに複数の文字がある場合、これは機能しません。

  2. ネストされた内包表記を使用してリストのリストをフラット化する（a）、リストの末尾から3未満の場合は_'x'_が追加された3つのグループにスライスされます。

    _>>> [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]_は、リストのギザギザのリストを平坦化します。

13
Mark Mikofski

これを試して

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

nは、'x'を挿入する要素の数の後です。

これは、変数iを初期化し、nと等しく設定することにより機能します。次に、whileがiの長さより短いときに実行されるlettersループを設定します。次に、iのインデックスlettersに'x'を挿入します。次に、n+1の値をiに追加する必要があります。 nだけでなくn+1を実行する必要がある理由は、lettersに要素を挿入すると、リストの長さが1つずつ拡張されるためです。

nが3で、'x'を挿入する例でこれを試してみると、次のようになります。

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

印刷する

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

これはあなたの期待する結果です。

8
michaelpri

forループでlist.insert()を使用するとメモリ効率が向上するようですが、1行で実行するために、均等に分割されたすべてのチャンクの最後に特定の値を追加することもできますリストのnthインデックスごとに分割します。

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

非常に簡単な方法：

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

また、チャンクにitertoolsドキュメントのgrouperレシピを使用することもできます。

2
Mike Graham

これは古いトピックですが、最も簡単で最も「Python的な」ソリューションimoがありません。 Mark Mikofski 's accepted answer のパート2の拡張にすぎません。間違いなく可読性が向上します（したがって、Python的になります）。

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

アイテムごとに新しい要素を追加したい。

これはどう ？

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

今です

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

Mark Mikofski の answer は機能しますが、スライスを割り当てることでより高速なソリューションがあります。

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

生産（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