寻找用Pythonic方式部分填充NumPy数组的方法

正确的直接赋值方法 #

你只需要同时指定两个维度的切片范围，一行代码就能完成：

psi[:, :500] = b

为什么你的尝试会报错？ #

你写的psi[:][:500]其实等价于(psi[:])[:500]：

1. psi[:]会返回原数组的一个视图，形状仍然是(3, 1000)
2. 接着的[:500]是对这个视图的第一维进行切片，但原数组第一维只有3个元素，所以结果还是形状(3, 1000)的数组
3. 这时候把形状为(3, 500)的b赋值进去，自然会因为维度不匹配触发ValueError

其他通用写法 #

如果想让代码适配更多场景（比如不确定b的具体维度时），可以用以下写法：

psi[:b.shape[0], :b.shape[1]] = b

或者用len()获取第一维长度：

psi[:len(b), :b.shape[1]] = b

内容的提问来源于stack exchange，提问作者bob.sacamento