Pythonで文字列を均等な長さに分割する方法【itertools活用】

このチュートリアルでは、指定した文字列を均等な長さの部分文字列に分割するプログラムを作成します。まず、具体的な例を見てみましょう。

入力例

string = 'Tutorialspoint'
each_part_length = 5

出力例

Tutor ialsp ointX

文字列の長さが各部分の長さで割り切れない場合、余った部分はfillvalueで指定した文字（ここでは「X」）で埋められます。

入力例

string = 'Tutorialspoint'
each_part_length = 6

出力例

Tutori alspoi ntXXXX

この結果を実現するために、itertoolsモジュールのzip_longestメソッドを使用します。

zip_longestメソッドとは

zip_longestメソッドは、イテレータを引数として受け取ります。また、文字列を分割する際の埋め合わせ用としてfillvalueを渡すこともできます。このメソッドは、同じ数の文字を含むタプルのリストを返します。

zip_longestは、渡されたイテレータの中で最も長いものが使い果たされるまで、反復ごとにタプルを返します。そして、そのタプルには各イテレータから指定した長さ分の文字が格納されます。

コード例1：5文字ずつ分割

# itertoolsモジュールをインポート
import itertools

# 文字列と分割する長さを初期化
string = 'Tutorialspoint'
each_part_length = 5

# 必要な数のイテレータを作成
iterator = [iter(string)] * each_part_length

# zip_longestを使って分割
result = list(itertools.zip_longest(*iterator, fillvalue='X'))

# タプルのリストを文字列に変換して出力
print(' '.join([''.join(item) for item in result]))

実行結果

上記のコードを実行すると、次の出力が得られます。

Tutor ialsp ointX

コード例2：6文字ずつ分割

# itertoolsモジュールをインポート
import itertools

# 文字列と分割する長さを初期化
string = 'Tutorialspoint'
each_part_length = 6

# 必要な数のイテレータを作成
iterator = [iter(string)] * each_part_length

# zip_longestを使って分割
result = list(itertools.zip_longest(*iterator, fillvalue='X'))

# タプルのリストを文字列に変換して出力
print(' '.join([''.join(item) for item in result]))

実行結果

上記のコードを実行すると、次の出力が得られます。

Tutori alspoi ntXXXX

まとめ

このように、itertoolsモジュールのzip_longestメソッドを使えば、文字列を任意の長さで均等に分割することができます。文字列が割り切れない長さの場合も、fillvalueを指定することで柔軟に対応できます。

このチュートリアルについて不明な点がある場合は、コメント欄でお気軽にお尋ねください。