[Python] 파일 이름 한번에 바꾸기

import os

file_path = './image/train' #파일이 존재하는 디렉터리 경로
file_names = os.listdir(file_path) #파일들을 리스트 형태로 바꿔줌
i = 1
for name in file_names:
    src = os.path.join(file_path, name)
    dst = str(i) + '.jpg'
    dst = os.path.join(file_path, dst)
    os.rename(src, dst)
    i += 1