Pythonで111222333といった、ある値が3つ連続した後次の数字が来るようなリストを作成する方法は?

Question

実現したいこと

＊超初心者な質問で申し訳ありません。
Pythonで
[1,1,1,2,2,2,3,3,3,4,4,4,5,5,5,6,6,6,・・・・]のように、ある値が3つ連続した後、次の数字が来るようなリストを作成したい場合、どういったコードになるでしょうか？

補足情報

以下の写真のようにlist(range())を使うと単純な連番を出すことができるとわかりました。

Answer 1

効率良い方法をご紹介します。
それなりに大きなリストを作らないと体感はできないかもですが。

>>> [n for n in range(1, 6) for _ in range(3)]
[1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5]

more-itertoolsを使うのも良い選択です。意味的に分かり易く、効率も良いコードが書けます。

>>> import more_itertools
>>> [*more_itertools.repeat_each(range(1, 6), n=3)]
[1, 1, 1, 2, 2, 2, 3, 3, 3, 4, 4, 4, 5, 5, 5]

さらに踏み込んで

必要なリスト長が前以て分からないときは次のように処理します。

import itertools
import more_itertools

it = more_itertools.repeat_each(itertools.count(start=1), n=3)

itはイテレータです。先頭から順に 1,1,1,2,2,2,3,3,3,... と値を取り出せます。

Answer 2

itertools の chain と tee で。

from itertools import chain, tee

rep = [*chain.from_iterable(zip(*tee(range(1, 11), 3)))]
print(rep)

Answer 3

rangeを使うのであれば、大きなサイズでなければ、下記が簡単です。

foo = sorted(list(range(1,6))*3)

Comments

[So0622]

sortedで並べ替えるってことですね！勉強になりました。ありがとうございます！