python - 如何在 Python 中初始化二维元素的一维数组？

Question

我想初始化一个包含 X 个二维元素的数组。例如，如果 X = 3，我希望它是 [[0,0]、[0,0]、[0,0]]。我知道 [0]*3 给出 [0, 0, 0]，但我如何为二维元素执行此操作？

Answer 1

试试这个:

m = [[0] * 2 for _ in xrange(3)]

在上面的代码中，将 3 视为矩阵中的行数，将 2 视为列数。 0 是您要用于初始化元素的值。请记住这一点:如果您使用的是 Python 3，请使用 range 而不是 xrange。

对于更通用的解决方案，请使用此功能:

def create_matrix(m, n, initial=0):
    return [[initial] * n for _ in xrange(m)]

对于问题的特殊情况:

m = create_matrix(3, 2)
print m
> [[0, 0], [0, 0], [0, 0]]

或者，如果您不介意使用 numpy，您可以使用 Mellkor 的回答中所示的 numpy.zeros() 函数。