画像集約フォルダからランダムに画像を抽出したい。

Question

前提・実現したいこと

10個新しいフォルダを作成し、別の画像集約フォルダからランダムに画像を5個ずつ取ってきて、作成したフォルダにコピーしたい。

発生している問題・エラーメッセージ

下記の方法だと、1つのフォルダ内で重複しない画像をランダムに取ってくるようになると思うのですが、これを今回新しく作成したフォルダ内（51‐60まで）で重複しないようにしたいです。
方法がわからず、どう検索したらいいかもわかりません。

該当のソースコード

import os
import glob, random
import shutil

start_num = 51
end_num = 61
new_dir_path = "‐new_dir_path-"

for i in range(start_num, end_num):
    os.mkdir(new_dir_path+str(i))
    photo_album_path = "-photo_album_path-"
    photo_album_inside = os.listdir(photo_album_path)
    chosen_photos = random.sample(photo_album_inside, 5)
    
    for chosen_photo in chosen_photos:
        chosen_photo_path = -photo_album_path-"+chosen_photo
        new_chosen_photo_path = chosen_photo_path.replace("photo_album",str(i))
        shutil.copy(chosen_photo_path, new_chosen_photo_path)

試したこと

補足情報（FW/ツールのバージョンなど）

Answer 1

random.sampleで50個とって、5個ずつに分けましょう。

>>> import random
>>> sample = random.sample(list(range(200)), 50)
>>> print(sample)
[76, 182, 148, 24, 106, 58, 5, 115, 85, 195, 158, 191, 41, 53, 149, 78, 12, 181, 40, 22, 131, 9, 31, 118, 107, 133, 19, 26, 91, 129, 1, 147, 48, 166, 55, 164, 168, 161, 140, 198, 172, 187, 36, 92, 72, 119, 138, 170, 62, 4]
>>> for i in range(0, 50, 5):
...     print(sample[i:i+5])
...
[76, 182, 148, 24, 106]
[58, 5, 115, 85, 195]
[158, 191, 41, 53, 149]
[78, 12, 181, 40, 22]
[131, 9, 31, 118, 107]
[133, 19, 26, 91, 129]
[1, 147, 48, 166, 55]
[164, 168, 161, 140, 198]
[172, 187, 36, 92, 72]
[119, 138, 170, 62, 4]

Comments

[Haru_]

ありがとうございます。わかりやすく、すぐに解決できました。

Answer 2

こんな感じでしょうか．

#省略

new_chosen_photo_path_set = set() #新しく作成するファイルpathの集合
dict_ = {} #元のファイルpathと新しく作成するファイルpathのdict

while len(new_chosen_photo_path_set) < 10:
  # 省略
  chosen_photo = random.choice(photo_albam_inside)
  chosen_photo_path = #省略
  new_chosen_photo_path = #省略
  new_chosen_photo_path_set.add(new_chosen_photo_path)
  dict_[new_chosen_photo_path] = chosen_photo_path

for new, old in dict_.items():
  shutil.copy(old,new)

setを使えば重複を避けれます．

Comments

[Haru_]

ご回答ありがとうございます。当方、超初心者のため他の方をベストアンサーにさせていただきましたが、今後setに関しても勉強していければと思います。迅速なご回答感謝いたします。