4分割された複数組みの画像を一括で結合して元の画像にしたい．

Question

最近，画像処理にpythonを用いており，分からない事項があったため質問させていただきます．

Google colabratoryで，pythonを使用しています．
現在，以下のようにcをタイル状に結合して元の画像にしています．

from google.colab import drive 
drive.mount('/content/drive')

from PIL import Image
 
data_root = "/content/drive/My Drive/image_div/"

# 4つの画像を開く
im1 = Image.open(data_root + "1234_1.jpg")
im2 = Image.open(data_root + "1234_2.jpg")
im3 = Image.open(data_root + "1234_3.jpg")
im4 = Image.open(data_root + "1234_4.jpg")

# 4つの画像を結合
def get_concat(im1, im2, im3, im4):
    dst = Image.new('RGB', (im1.width + im2.width, im1.height + im3.height))
    dst.paste(im1, (0, 0))
    dst.paste(im2, (im1.width, 0))
    dst.paste(im3, (0, im1.height))
    dst.paste(im4, (im1.width, im1.height))
    return dst

# 結合した画像を保存
get_concat(im1, im2, im3, im4).save(data_root + "image_cont/1234.jpg")
コード

分割された画像ファイル（結合したい画像）は以下のように，ディレクトリにまとまっています．

image_div
├ 1234_1.jpg
├ 1234_2.jpg
├ 1234_3.jpg
├ 1234_4.jpg
├ 1235_1.jpg
├ 1235_2.jpg
├ 1235_3.jpg
├ 1235_4.jpg
├ 1236_1.jpg
└ ...

4分割された画像のファイル名は組み合わせ毎に連番になっていて，後ろにも_1, _2, _3, _4と連番が振られています．結合して復元した画像は，そのファイル名前の前半部分（コード内で1234）をファイル名として保存しています．

少ない枚数であれば，
上のコード内の4分割された画像のファイル名前半部分，結合して復元した画像のファイル名の部分（コード内で1234）を手動で入力してからコードを実行していますが，数百枚の画像に対して同様の処理をする必要があり，一括で効率よく行いたいと考えています．

つまり，Image.openの引数で指定している，読み込みたい4分割された画像（結合したい画像）のファイル名前半部分と，saveの引数で指定している，結合して出来上がった画像の保存したいファイル名は同じであるため（コード内で1234），その部分の指定を自動でおこない，画像の読み込み・結合・保存を，ディレクトリ内にある同様の多数の画像に対して一括にできるようにしたいと考えています．

初めてのプログラミングに関する質問で，稚拙な説明で大変恐縮ですが，
ご助力いただきたいです．よろしくお願いいたします．

Answer 1

以下の方法を使えば、名前の一覧を得ることができます。

>>> import os
>>> import glob
>>> jpegs = glob.glob('image_div/*.jpg')
>>> print(jpegs)
['image_div\1234_1.jpg', 'image_div\1234_2.jpg', 'image_div\1234_3.jpg', 'image_div\1234_4.jpg', 'image_div\1235_1.jpg', 'image_div\1235_2.jpg', 'image_div\1235_3.jpg', 'image_div\1235_4.jpg', 'image_div\1236_1.jpg', 'image_div\1236_2.jpg', 'image_div\1236_3.jpg', 'image_div\1236_4.jpg']
>>> concatnames = {os.path.basename(j)[:4] for j in jpegs}
>>> print(concatnames)
{'1236', '1234', '1235'}

Comments

[Showthekey37]

ppaulさま，ありがとうございます．
ppaulさまの紹介してくださった方法を組み合わせることで解決いたしました．

Answer 2

回答者さまのおかげで解決いたしました．
ビギナーなので少し複雑なコードかもしれませんが，最終的に以下のコードで動作しました．

import glob
from PIL import Image

#GoogleDriveディレクトリ移動
%cd "/content/drive/My Drive/image_div" 

# ディレクト内の画像リスト表示＆並べ替え
jpegs = glob.glob('*.jpg')
jpegs.sort()

for start in range(0, len(jpegs), 4):
    # ４ずつ抽出
    divjpegs = jpegs[start : start + 4]
    
    # 4つの画像を開く
    im1 = Image.open(divjpegs[0])
    im2 = Image.open(divjpegs[1])
    im3 = Image.open(divjpegs[2])
    im4 = Image.open(divjpegs[3])

    # 4つの画像を結合
    def get_concat(im1, im2, im3, im4):
        dst = Image.new('RGB', (im1.width + im2.width, im1.height + im3.height))
        dst.paste(im1, (0, 0))
        dst.paste(im2, (im1.width, 0))
        dst.paste(im3, (0, im1.height))
        dst.paste(im4, (im1.width, im1.height))
        return dst
    
    concatnames = {os.path.basename(j)[:4] for j in divjpegs}
    concatnames_str = str(concatnames)
    concatnames_str2 = concatnames_str.replace('{', '')
    concatnames_str3 = concatnames_str2.replace('}', '')
    concatnames_str4 = concatnames_str3.replace("'", '')


    # 結合した画像を保存
    get_concat(im1, im2, im3, im4).save(concatnames_str4 + ".jpg")
コード