numpy，用数组替换列
提问于 2019-04-12 07:27:09
我试图用相同长度的新数组替换一个或几个列。
a = np.array([[1,2,3],[1,2,3],[1,2,3]])
b = np.array([[0,0,0])
a[:, 0] = b我得到了一个ValueError: could not broadcast input array from shape (3,1) into shape (3)错误。但是，当b有多个列时，这是可行的。
a = np.array([[1,2,3],[1,2,3],[1,2,3]])
b = np.array([[0,7],[0,7],[0,7]])
a[:, 0:2] = b
array([[0, 7, 3],
[0, 7, 3],
[0, 7, 3]])如何用另一个数组有效地替换列？
谢谢
J
回答 2
Stack Overflow用户
回答已采纳
发布于 2019-04-12 07:33:16
您的b形状不正确。如果只想替换一个列，则应该将一个普通的一维数组传递给它：
a = np.array([[1,2,3],[1,2,3],[1,2,3]])
b = np.array([0,0,0])
a[:, 0] = b
a返回：
array([[0, 2, 3],
[0, 2, 3],
[0, 2, 3]])Stack Overflow用户
发布于 2019-04-12 07:31:53
如果您像使用a[:, 0:2] = b一样使用以下内容，那么您的示例将工作得很好。[:, 0:1]实际上只是第一列
a = np.array([[1,2,3],[1,2,3],[1,2,3]])
b = np.array([[0],[0],[0]])
a[:, 0:1] = b
# array([[0, 2, 3],
# [0, 2, 3],
# [0, 2, 3]])页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/55654972
复制相关文章
![[答疑]人翻书，翻书的操作是在人上还是书上](https://ask.qcloudimg.com/http-save/yehe-6288414/lkuegum6ts.png)
腾讯云开发者
