Utils
랜덤으로 파일 추출하는 파이썬 스크립트
jinmc
2022. 4. 21. 18:32
반응형
# source directory 지정 target directory 지정
import os
import random
from shutil import copyfile
def main():
print("select source directory")
ldir = os.listdir()
ldir = [l for l in ldir if os.path.isdir(l)]
print(ldir)
source = input()
print("select target directory")
target = input()
ldir2 = os.listdir(source)
print(f"There are {len(ldir2)} items in source directory")
print("how many items do you want to select randomly?")
nums = input()
val = 0
try:
val = int(nums)
except ValueError:
print("Not number!")
return
print(f"You want to select {nums} in the directory")
selected = random.sample(ldir2, val)
print(f"{len(selected)} selected!!")
print("press enter to see")
input()
print(selected)
for fl in selected:
copyfile(f"./{source}/{fl}", f"./{target}/{fl}")
if __name__ == '__main__':
main()
반응형