为什么我不能为2-d numpy.ndarray建立索引并分配新值
为什么我不能为2-d numpy.ndarray建立索引并分配新值
提问于 2019-05-13 16:55:08
我想索引一个2dndarray并获取这个ndarray的一个子集。然后，我将这个子集ndarray分配给与这个子集ndarray具有相同形状的另一个ndarry。但是原始的ndarray没有改变。
这里有一个例子，赋值操作后mat没有改变。
我想要的是分配一个具有另一个ndarray.的subset of ndarray
import numpy as np
ma = np.array([1,2,3,4,5] * 5).reshape(5,5)
mask = np.array([False, True, False, True, True])
sub = np.array([[100,101, 102],
[103, 104, 105],
[106,107,108]])
ma[mask][:,mask] = sub
print(ma)我所期望的是：
array([[1, 1, 3, 1, 5],
[1, 100, 1, 101, 102],
[1, 1, 3, 1, 5],
[1, 103, 3, 104, 105],
[1, 106, 3, 107, 108]])但是mat没有改变：
array([[1, 2, 3, 4, 5],
[1, 2, 3, 4, 5],
[1, 2, 3, 4, 5],
[1, 2, 3, 4, 5],
[1, 2, 3, 4, 5]])回答 2
Stack Overflow用户
发布于 2019-05-13 17:20:40
np.where(mask, 1, np.where(mask, 1, ma).T).Tnp.where(mask, 1, ma)使用掩码替换为columnsnp.where(mask, 1, np.where(mask, 1, ma).T)中的1转置结果，然后再次重复以掩码rowsnp.where(mask, 1, np.where(mask, 1, ma).T).T转置返回
输出
np.where(mask, 1, np.where(mask, 1, ma).T).T
array([[1, 1, 3, 1, 5],
[1, 1, 1, 1, 1],
[1, 1, 3, 1, 5],
[1, 1, 1, 1, 1],
[1, 1, 3, 1, 5]])Stack Overflow用户
发布于 2019-05-13 17:28:52
执行以下操作：
ma[np.ix_(mask, mask)] = sub页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/56108865
复制相关文章
相似问题
腾讯云开发者
