python - 如何在 python 3.7.0 中使用 os 模块移动具有相似名称的文件

Question

我想使用 python 3.7.0 中的 os 模块将具有相似名称的文件从一个目录移动到另一个目录。例如，我有名称为 A1B1、A1B2、A2B1、A2B2 等的文本文件。如何在名为 A1 的目录中移动名为 A1Bn (n=1,2,3...) 的文件，以及在名为 A2 的目录中移动名为 A2Bn (n=1,2,3...) 的文件。如何检查文件名。谢谢 ...

标签： python

Answer 1

用于os.listdir从文件夹中获取所有文件并os.rename移动文件。

import os


def file_reader():
    # get files
    for files in os.listdir("/path/to/your/files"):
        if files.endswith(".txt"):
            try:
                os.mkdir("A1")
                os.mkdir("A2")
            except FileExistsError:
                pass
            if files.startswith("A1"):
                os.rename(files, os.path.join("A1", files))
            if files.startswith("A2"):
                os.rename(files, os.path.join("A2", files))


if __name__ == "__main__":
    file_reader()