在python中创建2d数组？

Question

这是我试图创建2d矩阵的代码

m=4
tagProb=[[]]*(m+1)
count=0
index=0
for line in lines:
    print(line)
    if(count < m+1):
       tagProb[index].append(line.split('@@')[2].strip()) 
       count+=1
    if(count == m+1): // this check to goto next index
        count = 0
        index+=1
print(tagProb)  

我得到了o/p

[['0.0', '0.6', '0.05', '0.3', '0.05', '0.1', '0.0', '0.6', '0.0', '0.0', '0.1', '0.0', '0.0', '0.9', '0.0', '0.1', '0.0', '0.2', '0.7', '0.0', '0.1', '0.0', '0.9', '0.0', 0.0'], ['0.0', '0.6', '0.05', '0.3', '0.05', '0.1', '0.0', '0.6', '0.0', '0.0', '0.1', '0.0', .0', '0.9', '0.0', '0.1', '0.0', '0.2', '0.7', '0.0', '0.1', '0.0', '0.9', '0.0', '0.0'], '0.0', '0.6', '0.05', '0.3', '0.05', '0.1', '0.0', '0.6', '0.0', '0.0', '0.1', '0.0', '0.0','0.9', '0.0', '0.1', '0.0', '0.2', '0.7', '0.0', '0.1', '0.0', '0.9', '0.0', '0.0'] ]

附加所有值,列表具有相同的值.我怎么能避免这个？

Answer 1

您正在使用*列表,它有一个陷阱 - 它将列出对同一对象的大量引用.这对于像ints或tuples 这样的不可变类很好,但不适用于像mutable这样的变量list,因为更改其中一个对象会改变所有对象.看到:

>>> foo = [[]]*10
>>> foo[0].append(1)
>>> foo
[[1], [1], [1], [1], [1], [1], [1], [1], [1], [1]]

如果您不希望这种情况发生,那么避免它的标准方法是使用列表推导,它将使用新对象初始化列表:

>>> bar = [[] for _ in range(10)]
>>> bar[0].append(1)
>>> bar
[[1], [], [], [], [], [], [], [], [], []]

然而,这个问题在惯用Python中并不是很有用,因为初始化一个大型列表并不常见 - 这是一种C心态.(这并不是说它有时候不是正确的做法 - Python是多范式的!)

另外一点,你的代码并不好.forPython中的循环旨在处理对象的迭代,这样您就不必手动管理索引变量(index和count代码中的变量).最好改写如下:

import numpy as np
m = 4
tagProb = np.array(list(line.split("@@")[2].strip() for line in lines)))
tagProb = tagProb.reshape((m+1,-1)).T

说明:第一行定义tagProb为一维的numpy数组(具有大量线性代数函数的快速基于C的数组类型),其中所有值都在一行中.第二行将它强制转换为高度m+1和推断宽度的矩阵(注意它必须是正方形才能工作; None如果需要,可以用它填充)然后转置它.我相信这是你的迭代所做的,但它有点难以理解 - 让我知道你是否想要这样做.