使用可变对象初始化多维数组的更好方法

Question

我想创建一个多维数组并用 a 的副本初始化它 可变对象。这就是我到目前为止所拥有的：

import copy

def create_array(dimensions):
    dimensions = copy.deepcopy(dimensions)
    dimensions.reverse()
    a = [0] * dimensions[0]
    del dimensions[0]
    for d in dimensions:
        a = [copy.deepcopy(a) for _ in range(d)]
    return a

def create_array_mutable(dimensions, obj):
    a = create_array(dimensions)
    def set(x):
        if isinstance(x[0], list):
            for e in x:
                set(e)
        else:
            for i in range(len(x)):
                x[i] = copy.deepcopy(obj)
    set(a)
    return a

我想知道是否有更好的方法来做到这一点（没有副本和递归）？

Answer 1

怎么样：

import copy

def create_array_mutable(dims, obj):
  if len(dims) == 0:
    return copy.deepcopy(obj)
  else:
    return [create_array_mutable(dims[1:], obj) for i in xrange(dims[0])]

class C(object): pass

print create_array_mutable((2,3,4), C())

这将创建一个由 C 的唯一实例组成的 2x3x4 数组。

解决方案仍然是递归的，但我认为递归非常适合这个问题。