1. 程式人生 > >python-批量建立指定名稱的資料夾

python-批量建立指定名稱的資料夾

前言:在用深度學習的方法做目標分類時,會遇到建立資料集的問題。比如說需要建立200類,每類目標存放500張圖片。這樣就需要首先建立200個不同名稱的資料夾。然而,如果一個個建立顯然要不得,因此,就有了以下的程式碼:

1.批量建立N個資料夾:

import os, sys
def MkDir():
    path = './file/'#建立檔案路徑
    i = 0
    for i in range(200): #建立檔案個數
        file_name = path + str(i)
        os.mkdir(file_name)
        i=i+1
MkDir()

2.批量建立N個指定名稱的資料夾:

import os, sys
path1 = './dataset'  #指定名稱資料夾所在路徑
path2 = './file/'    #新建資料夾所在路徑

def MkDir():
    dirs = os.listdir(path1)

    for dir in dirs:
        file_name = path2 + str(dir)
        os.mkdir(file_name)

MkDir()