Python-根据txt中的文件名批量移动文件

最近遇到一个问题，一个有7000多张图片的数据集，需要按照已给出的train.txt、test.txt、val.txt中的图片名字把数据集分成三个子数据集。使用代码如下所示（简单几行超好用）。

import shutil

if __name__ == '__main__':
    file_object = open('/home/jiao/Pictures/image/zz.txt')
    try:
        for line in file_object:
            # print(line)
            shutil.move('/home/jiao/Pictures/image/'+line.rstrip('\n')+'.jpg', "/home/jiao/Pictures/zz")
    finally:
        file_object.close()

一个小示例：
image文件夹中有若干图片，zz.txt记录着需要移动的图片名字，将txt中记录的图片移动到新的zz文件夹中。


程序运行后