python - 替换多维数组的元素

Question

我试图用列表中的元素替换 29 x 1 矩阵的一些元素。

import numpy as np

initAmount = np.array([[0 for i in range(29)]])
initialAmount = np.ndarray.transpose(initAmount)
ind = [0.04, 0.02, 0.03]
#ind = [1,2,3]
initialAmount[0,0] = float(ind[0])
initialAmount[1,0] = float(ind[1])
initialAmount[2,0] = float(ind[2])
print(initialAmount)

不幸的是，这并没有像预期的那样工作。运行代码后的初始金额应为 [[0.04],[0.02],[0.03],[0],...]，而不是 [[0],[0],[0]....]，其中是我得到的结果。当我的列表 ind = [1,2,3] 时，代码工作正常。因此，我假设精度存在错误，但我不知道如何解决这个问题。

任何帮助将不胜感激。

Answer 1

只需使用内置的 numpy.zeros 来创建数组，它采用 dtype 和 shape 参数，极大地简化了您的尝试完成:

>>> init = np.zeros((29, 1), dtype=float)
>>> ind = [0.04, 0.02, 0.03]
>>> init[0,0] = ind[0]
>>> init[1,0] = ind[1]
>>> init[2,0] = ind[2]
>>> init
array([[ 0.04],
       [ 0.02],
       [ 0.03],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ],
       [ 0.  ]])
>>>

注意，默认情况下np.zeros使用float dtype。但明确地表达出来并没有什么坏处。