Python的numpy矩阵列表表现得很奇怪

时间:2013-02-23 22:28:47

标签: python arrays list matrix numpy

我正在尝试使用numpy矩阵列表,并且遇到了烦人的问题。

假设我从十个2x2零矩阵的列表开始

para=[numpy.matrix(numpy.zeros((2,2)))]*(10)

我访问这样的单个矩阵

para[0]
para[1]

等等。到目前为止一切都很好。

现在,我想仅修改第二个矩阵的第一行,保持所有其他矩阵不变。所以我这样做

para[1][0]=numpy.matrix([[1,1]])

第一个索引指向列表中的第二个矩阵,第二个索引指向该矩阵中的第一行,将其替换为[1,1]。

但奇怪的是,这个命令将列表中所有十个矩阵的第一行更改为[1,1]而不是像我想要的那样只改变第二行。是什么给了什么?

1 个答案:

答案 0 :(得分:5)

当您将初始列表乘以10时,最终会得到10个numpy数组的列表,这些数组实际上是对相同底层结构的引用。修改一个会修改所有这些,因为事实上只有一个numpy数组,而不是10个。

如果您需要证明,请在REPL中查看此示例:

>>> a = numpy.zeros(10)
>>> a = [numpy.zeros(10)]*10
>>> a[0] is a[1]
True
>>> 

is运算符检查两个对象实际上是否相同(如果它们的值相等则不一样)。

你应该做的是使用列表推导来生成初始数组而不是乘法,如下所示:

para=[numpy.matrix(numpy.zeros((2,2))) for i in range(10)]

这将调用numpy.matrix()十次而不是一次,并生成10个不同的矩阵。