使用 Python 根據文件名創建文件夾

Question

我在一個文件夾中有大約一百萬個文件，我想使用 Python 根據部分文件名將它們組織到文件夾中。 下面包括一些文件名示例。 我想根據第一個“_”之后和第二個“_”之前的信息為每種文件類型創建一個文件夾。

19930813_10-Q_edgar_data_60512_0000060512-94-000005_1

19931007_10-K-A_edgar_data_60512_0000060512-94-000006_1

所以對於第一個文件，我想創建一個名為“10-Q”的文件夾，第二個是“10-K-A”。 我知道有幾個類似的查詢，但它們主要用於其他編程語言，而不是 Python。 謝謝您的幫助。

Answer 1

試試這個代碼，希望它能工作：

import os
import shutil

folder_path = 'The folder containing files'
file_types = []
for file in (os.listdir(folder_path)):
    file_type_part_name_by_ = file.split('_')
    file_type = file_type_part_name_by_[1]
    if file_type not in file_types:
        file_types.append(file_type)
        os.mkdir(file_type)
        shutil.move(file, file_type)
    else:
        shutil.move(file, file_type)

---