Python 2次元リストの並べ替え

Question

入力された文字列の高さを幅を変更して出力するプログラムを書いています。
例えば下記のコードで
3 4 2
abcd
efgh
ijkl

と入力されると、1行目3つ目の「2」の入力を新たな文字幅として受け、
ab
cd
ef
gh
ij
kl

と出力されるプログラムが書きたいです。

しかし、新たな文字幅に「7」など割り切れない数字を入力した際に、

abcdefg
hijkl

というように表示されません。
下記の書き方では問題あるでしょうか？

import numpy as np

#それぞれ高さ、幅、変更後の幅を入力
h, w, x = map(int, input().split())

display = []
#入力された文字列をリストに格納
for i in range(h):
    sentence = list(input())
    display.append(sentence)

#2次元リストを1次元に直す
f_display = np.ravel(display)
#1次元リストを新たな高さ、幅で格納
n_display = f_display.reshape(int(H*W/X),-1 )
#新たな文字列を出力
for n in n_display:
    print(*n, sep="")

教えていただければと思います。よろしくお願い致します。

Answer 1

割り切れない幅を想定する場合、reshapeは向かないように思います。
今回の場合スライスを使うのがシンプルで簡単なのではないでしょうか。

while len(f_display) != 0:
    d, f_display = f_display[:x], f_display[x:]
    print(*d, sep="")

f_displayの型は一次元のndarrayでも組み込みのリストでも動きます。

書いてみた

おもしろそうだったので。

import sys

_, _, w = map(int, input().split())

s = ''.join(map(str.rstrip, sys.stdin))
while s:
    line, s = s[:w], s[w:]
    print(line)

Comments

[Justin04]

ここまで丁寧に解説していただき感謝です。よく考えればリストにする必要はなく、文字列として扱えば良いんですね…。非常に勉強になりました！