python - 奇数が繰り返される整数のリスト

Question

0連続するN整数のリストが与えられます。ここNで、は偶数です。たとえば、入力リストの奇数を繰り返す別のリストを作成するA = [0 , 1 , 2 , 3 , 4]ためのリスト内包表記、つまりタイプを探しています。[a for a in A]B = [0 , 1 , 1 , 2 , 3 , 3 , 4]A

Answer 1

リスト内包表記を使用した（やや醜い）ソリューションは次のとおりです。

a = range(5)
list(itertools.chain(*[[x, x] if x % 2 == 1 else [x] for x in a]))
# => [0, 1, 1, 2, 3, 3, 4]

リスト内包表記は、サブリストのリストを作成します。

[[x, x] if x % 2 == 1 else [x] for x in a] # => [[0], [1, 1], [2], [3, 3], [4]]

この呼び出しは、Pythonでリストitertools.chainをフラット化する多くの方法の1つにすぎません。

Answer 2

itertools.chain(*[[i] * (i % 2 + 1) for i in xrange(N + 1)])