x個ずつリストの先頭に値を追加したい

Question

以下のようにlistAとlistBを結合してlistCを作成したいのですが、良い案が浮かばず、ご教授いただきたいです。

listA=[[A,B],[C,D],[E,F],[G,H],[I,J],[K,L]]
listB=[[あ,い],[う,え]]

listC=[[あ,い,A,B],[あ,い,C,D],[あ,い,E,F],[う,え,G,H],[う,え,I,J],[う,え,K,L]]

listAのリスト3つずつに対して、listBをそれぞれ結合するような感じです。

listAとlistBで多重ループをして実現しようとしましたが、うまくいかず、本来[あ,い]を付与したくない[G,H]にも[あ,い]が付与されてしまい上手くいきません。

是非ご教授いただけませんでしょうか。
よろしくお願いします。

追記

listAの要素数は今回6要素ですが都度変わります。ただ、変わったとしても3の倍数ですので、必ず割り切れます。

Comments

[suomo]

3つずつ、というのは、今回の場合6要素のリストAに紐づけたいリストBが2要素しかないので、6/2=3で結果的に3要素ずつなのか、
それとも3要素は絶対に守りたい条件なのかどちらでしょうか？
(今回は割り切れるので問題ありませんが、例えばリストAが7要素でリストBが2要素のときに、前半4つと後半3つに分割するのか、あるいは3個ずつつけていって余ったリストAの最後の1要素には、最初からループしてまたリストBの1要素目をつけたいのか、といった違いが生じるためです。明確にされたほうが的確な回答が得られると思います。)

Answer 1

複雑な事しなくても良かった。

listBの各要素を3回繰り返すリストを作り、
zipでlistAと掛け合わせます。

listA = [["A","B"],["C","D"],["E","F"],["G","H"],["I","J"],["K","L"]]
listB = [["あ","い"],["う","え"]]
listC = [b + a for a, b in zip(listA, (b for b in listB for _ in range(3)))]

# for a,b in zip(listA, (b for b in listB for _ in range(3)):
#     print(b + a)

※リスト内包表記に変更。元のコードはコメントアウト。

Comments

[jun68ykt]

「その手があったか！」という感じです。質問を見て「まずは listAを chunk だな」と（先入観で）思い込んでしまうと、これはなかなか思いつきません。

[ruto_07]

リスト内包表記が分かりやすく、こちらで実装させていただきたいと思います。
ご回答ありがとうございました。

[teamikl]

＞jun68yktさん

いろいろな方法があるのはわかっていましたが、
実際にどういう過程で書かれたコードなのか、考えてみて興味深かったです。

問題の A の要素3つに対して B をそれぞれ～
を文面通り忠実にコードに落とすなら、確かにlistAを分割は自然な流れですね。

組み合わせ=>zip, 繰り返し=>repeat を漠然とイメージしていて、
zipを利用するにあたり、リストの長さを揃えるのを考えました。
chunkは寧ろ listA, listB の構築に。

itertoolsをアピールしたくて書いたコードが、
整理してると不要になったのは自分でも意外でした:)

Answer 2

こんにちは
面白い問題なので参加します。（chunkは、stackoverflow の回答を拝借しました。）

# FYI: https://stackoverflow.com/a/312464
def chunks(lst, n):
    """Yield successive n-sized chunks from lst."""
    for i in range(0, len(lst), n):
        yield lst[i:i + n]


if __name__=='__main__': 

    listA = [['A','B'],['C','D'],['E','F'],['G','H'],['I','J'],['K','L']]
    listB = [['あ','い'],['う','え']]

    chunked_list = list(chunks(listA, 3)) # => [[['A', 'B'], ['C', 'D'], ['E', 'F']], [['G', 'H'], ['I', 'J'], ['K', 'L']]]

    ziped_itr = zip(listB, chunked_list) # => [(['あ', 'い'], [['A', 'B'], ['C', 'D'], ['E', 'F']]), (['う', 'え'], [['G', 'H'], ['I', 'J'], ['K', 'L']])] 

    result = [[[*x[0], *y] for y in x[1]] for x in ziped_itr] # => [[['あ', 'い', 'A', 'B'], ['あ', 'い', 'C', 'D'], ['あ', 'い', 'E', 'F']], [['う', 'え', 'G', 'H'], ['う', 'え', 'I', 'J'], ['う', 'え', 'K', 'L']]]

    result = sum(result, [])
    
    print(result)  # => [['あ', 'い', 'A', 'B'], ['あ', 'い', 'C', 'D'], ['あ', 'い', 'E', 'F'], ['う', 'え', 'G', 'H'], ['う', 'え', 'I', 'J'], ['う', 'え', 'K', 'L']]

• 動作確認用 Repl.it: https://repl.it/@jun68ykt/Q250503

Comments

[ruto_07]

いただいたご回答も参考になりましたが、もっと簡単に実現できる方法がありましたので、そちらで実装したいも思います。
ご回答ありがとうございました。

[jun68ykt]

どういたしまして。teamiklさんのベストアンサーのほうにもコメントしましたが、listB の要素を3個ずつにするというアプローチは思いつきもしませんでした。興味深いご質問、ありがとうございました。

Answer 3

more-itertoolsというライブラリにchunkedというメソッドがあります。これは与えたリストから指定の個数ずつまとめて取り出してくれるというものです。

例えば、

from mote-itertools import chunked

for chunk in chunked(listA, 3):
    ...

とすると、一回目のループでchunkとして[[A,B],[C,D],[E,F]]が取り出され、二回目のループで残りの3つ[[G,H],[I,J],[K,L]]が取り出されます。

今回の場合、リストBの要素数はリストAに併せて増減する想定だと思いますので、以下のようなコードでいかがでしょうか。

from mote-itertools import chunked

listC = []
for as, b in zip(chunked(listA, 3), listB):
    new_as = [b + elm for elm in as] #asの各要素の先頭にbを付け足したリストを新たに作る
    listC.extend(new_as) #listCに上の行で作ったリストを順に連結していく

print(listC)

Comments

[ruto_07]

いただいたご回答も参考になりましたが、簡単に実現できる方法がありましたので、そちらで実装したいも思います。
chunkedというメソッドはまた別の機会に使ってみたいと思います。
ご回答ありがとうございました。

[suomo]

せっかく追記依頼をお願いしておきながら、結局簡便な案を提示できませんでした。chunkedの便利なところは、余りがあってもいい感じにあまりを一つの要素として残してくれるところです。(例えば8要素のリストAに対してchunked(listA, 3)とすると、３＋３＋２に分けてくれます)

使いどころがありましたら是非に。

Answer 4

追記: よりシンプルに書けたので別回答で投稿します

シンプルに、Bの要素を３回回し、Aを一個ずつ読み出し
iter()化してnext()で読みだすところがポイント。

listA = [["A","B"],["C","D"],["E","F"],["G","H"],["I","J"],["K","L"]]
listB = [["あ","い"],["う","え"]]
listC = []


iterA = iter(listA)
for b in listB:
    for _ in range(3):
        a = next(iterA)
        listC.append(b + a)

from pprint import pprint
pprint(listC)

itertools 使った方法（関数プログラミングとか好きな人向け

説明追記:
listB の要素に対して、repeatで3回繰り返したリストを作り
chain.from_iterable で一段階フラットにします。
[["あ", "い"], ["あ", "い"], ["あ", "い"], ["う", "え"], ["う", "え"], ["う", "え"]]
と、listA と同じ長さのリストになるので、zipで掛け合わせます。

from itertools import chain, repeat

def chunk(size, seq):
    return list(zip(*[iter(seq)]*size))

listA = list(map(list, chunk(2, "ABCDEFGHIJKL")))
listB = list(map(list, chunk(2, "あいうえ"))) # [["あ", "い"], ["う", "え"]] を生成

def genListC(listA, listB):
    for a, b in zip(listA, chain.from_iterable(repeat(b,3) for b in listB)):
        yield b + a

listC = list(genListC(listA, listB))

from pprint import pprint
pprint(listC)

追記: ※ listをたくさん使ってますが、元のデータに合わせる為にいれただけなので、
最後以外はなくてもロジック自体は動きます。

Answer 5

listA = [["A", "B"], ["C", "D"], ["E", "F"], ["G", "H"], ["I", "J"], ["K", "L"]]
listB = [["あ", "い"], ["う", "え"]]

ans = [listB[(i // 3)] + listA[i] for i in range(len(listA))]
print(ans)
print()

listA = [
    ["A", "B"], ["C", "D"], ["E", "F"], ["G", "H"], ["I", "J"], ["K", "L"],
    ["a", "b"], ["c", "d"], ["e", "f"], ["g", "h"], ["i", "j"], ["k", "l"]
]
ans = [listB[(i // 3) % len(listB)] + listA[i] for i in range(len(listA))]
print(ans)

実行例：

Comments

[teamikl]

nice approach
可読性は少し落ちますが、enumerate でも行けますね
listAを複数書きたくない＆短縮で。

[listB[i//3]+a for i,a in enumerate(listA)]

[katoy]

↑のコードは短くて良いですね。

Answer 6

3個ずつというと、下記のような格好でしょうか。

listA=[["A","B"],["C","D"],["E","F"],["G","H"],["I","J"],["K","L"]]
listB=[["あ","い"],["う","え"]]

listC = []
l = len(listA)/3
for a in range(len(listA)):
    listC.append(listB[int(a/3)]+listA[a])
print(listC)

[['あ', 'い', 'A', 'B'],
['あ', 'い', 'C', 'D'],
['あ', 'い', 'E', 'F'],
['う', 'え', 'G', 'H'],
['う', 'え', 'I', 'J'],
['う', 'え', 'K', 'L']]

Comments

[ruto_07]

いただいたご回答も参考になりましたが、もっと簡単に実現できる方法がありましたので、そちらで実装したいも思います。
ご回答ありがとうございました。