在 python 中创建/制作目录（复杂）

Question

我正在尝试创建一堆可以将文件复制到其中的目录/子目录。我正在使用Python，但我似乎找不到一个好的方法来做到这一点。我有一条主路，我将从中分支出来。之后，我有 Weights 和 No_Weights。男性和女性跟随。在每个男性和女性文件夹中，我都有每个种族（白种人、非裔美国人、亚洲人、西班牙裔、印度人、其他人、未知）。在每个文件夹中，我的年龄范围从 20 岁以下一直到 70 岁以上（B20、20、30、40、50、60、70）。

我尝试生成所有路径，因此我只需调用 mkdir 大约 50 次，但这大约有 150 行代码（几乎）。

有没有一种简单的方法可以创建所有这些文件夹而无需手动完成？

Answer 1

import itertools
import os

dirs = [["Weights", "No_Weights"],
        ["Male", "Female"],
        ["Caucasian", "African-American", "Asian", "Hispanic", "Indo", "Other", "Unknown"], 
        ["B20", "20", "30", "40", "50", "60", "70"]]

for item in itertools.product(*dirs):
    os.makedirs(os.path.join(*item))

itertools.product() 将构建所有可能的路径变体，然后 os.path.join() 将使用适合您平台的正确语法将子路径连接在一起。

编辑：需要 os.makedirs() 而不是 os.mkdir()。只有前者才会构建完整路径中的所有中间子目录。

Answer 2

这个例子应该可以帮助您入门：

import itertools
import os.path

ROOT = r'C:\base\path'

sex = ('male', 'female')
ethnicity = ('Caucasian', 'African-American', 'Asian')
ages = ('B20', '20', '30')

for path in itertools.product(sex, ethnicity, ages):
    print os.path.join(ROOT, *path)

itertools 模块是您的朋友：
http://docs.python.org/library/itertools.html#itertools。产品

Answer 3

只需做这样的事情：

main = 'somedir'
weight = ['weights', 'No_weights']
ethnicity = ['Caucasian', #the rest]
ages = ['B20'] +  range(20, 71, 10)

for w in weights:
    os.mkdir(os.path.join(main, w)
    for e in ethnicity:
        os.mkdir(os.path.join(main, w, e))
        for a in ages:
            os.mkdir(os.path.join(main, w, e, a))

这应该会为你解决......

Answer 4

有几个嵌套的 for 循环，然后每个循环都有 os.mkdir 。使用 os.path.join 将目录路径连接在一起。

像这样的东西：

loop weights
    mkdir weight
    loop sexes
        mkdir weights + sex
        loop ethnicities
            mkdir weights + sex + ethnicity
            loop ages
                mkdir weights + sex + ethnicity + age

这里循环只是一个普通的 for 循环：

for weight in ('weights', 'no_weights'):

mkdir is os.mkdir

'+' is os.path.join:

os.mkdir(os.path.join(weights, sex, ethnicity, age))

编辑： dir_util 可能在这里有用，所以你不必创建每个子目录：

http://docs.python.org/release/2.5。 2/dist/module-distutils.dirutil.html

loop weights
    loop sexes
        loop ethnicities
            loop ages
                mkpath weights + sex + ethnicity + age

Answer 5

os.makedirs 可以提供帮助 - 它使所有中间目录一直到您指定的“叶”目录。

另一个问题（生成所有“A 列中的一个，B 列中的一个，...”组合）最好作为“混合基数计数”问题来解决 - 粗略地说，s/thing like...：

choices = [ ['Weights', 'Noweights'],
            ['Male', 'Female'],
            ['Caucasian', 'AfricanAmerican', ...
            ...
          ]
Ls = [len(x) for x in choices]
ct = [0] * len(Ls)

while True:
    p = [c[i] for i, c in zip(ct, choices)]
    os.makedirs(os.path.join(p))
    n = -1
    while n > -len(Ls):
      ct[n] += 1
      if ct[n] < Ls[n]: break
      ct[n] = 0
      n -= 1
    else:
      break

< code>itertools.product 是一种现代而简洁的方法，用于生成所有“A 列中的一个，等等”选择，以及我在生产软件中的建议 - 只是：

for p in itertools.product(*choices):
    os.makedirs(os.path.join(p))

可以替换所有上述代码（ ！）。我认为还值得了解“混合基计数”较低抽象级别的方法，因为它在许多情况下都很方便（包括使用 Python 版本 <2.6;-)。