python - 如何分配 Numpy 数组的特定部分？

Question

这就是我想要做的：

a = [[1, 2],[3,4]]
b = np.zeros(shape = (2,2))
b[:, 1:] = a[:,1:]

但我收到此错误消息：

TypeError：列表索引必须是整数或切片，而不是元组

标签： pythonarraysnumpytypeerror

Answer 1

正如@MattDMo 所提到的，您正在尝试在 2D 列表上使用 numpy 切片。您可以通过简单地将列表转换为 numpy 数组来解决此问题

a = np.array([[1, 2],[3,4]])
b = np.zeros(shape = (2,2))
b[:, 1:] = a[:,1:]