python3 任意の数で配列を分割

Question

任意の場所で配列を分割したいです。

l=[[1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6]]

y=2

for i in l:
	#print(i)
	print(list(zip(*[iter(i)]*int(y))) )

"""
[(1, 2), (3, 4), (5, 6)]
[(1, 2), (3, 4), (5, 6)]
[(1, 2), (3, 4), (5, 6)]
"""

2つに分割したいのですが、3つづつになってしまいます。

y=3なら、3分割したいと思ってます。

#y=2の時　期待する出力

[[1,2,3],[4.5,6]]
[[1,2,3],[4.5,6]]
[[1,2,3],[4.5,6]]

#y=3の時　期待する出力

[[1,2],[3,4],[5.6]]
[[1,2],[3,4],[5.6]]
[[1,2],[3,4],[5.6]]

yが分割するlen要素/y＝0に　ならない時
[4分割などの場合]
は何もしません。

Comments

[LouiS0616]

『期待する結果』を質問に追記してください。

Answer 1

from more_itertools import chunked

l = [[1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6]]

print("------")
y = 2
for i in l:
    if len(i) % y == 0:
        print(list(chunked(i, y)))

print("------")
y = 3
for i in l:
    if len(i) % y == 0:
        print(list(chunked(i, y)))

print("------")
y = 4
for i in l:
    if len(i) % y == 0:
        print(list(chunked(i, y)))

参考情報

• python2でリストをn個ずつのサブリストに分割する

http://y0m0r.hateblo.jp/entry/20140405/1396674065

Comments

[退会済みユーザー]

コンソールで pip install more-itertools としてからですね。
このモジュールは知りませんでした。

Answer 2

単純なスライスは都合悪いんですか？

l = [[1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6], [1, 2, 3, 4, 5, 6]]

y=2

for i in l:
    step = len(i) // y
    print([i[step*idx: step*(idx+1)] for idx in range(0, y)])
    # print([i[idx:idx+step] for idx in range(0, len(i), step)]  # こちらでも同じ。コメントを受けて追記

Comments

[LouiS0616]

print([i[idx:idx+step] for idx in range(0, len(i), step)]) の方がより素直な気もします。

[退会済みユーザー]

回答に追記しておきました。

Answer 3

分割幅を計算すれば良いだけでは？

my_list = [
    [1, 2, 3, 4, 5, 6],
    [1, 2, 3, 4, 5, 6],
    [1, 2, 3, 4, 5, 6]
]

y = 2
for row in my_list:
    print(list(zip(*[iter(row)]*(len(row)//y))))

実行結果 Wandbox

[(1, 2, 3), (4, 5, 6)]
[(1, 2, 3), (4, 5, 6)]
[(1, 2, 3), (4, 5, 6)]

とは言え流石に煩雑になってきたので、関数などに整理する必要がありそうですね。