我是一个天真的numpy用户,需要你的帮助以解决以下问题:我想用第三个数组替换多维数组的一些元素,这些元素少于第二个数组;例如.:
x = np.arange(16).reshape((2,8))
# x = np.array([[ 0,1,2,3,4,5,6,7],# [ 8,9,10,11,12,13,14,15]])
和
y = np.array([[2],[13]])
# y = np.array([[ 2],[13]])
现在,找出x大于y的位置,以及x中是否存在至少一个True> y数组,计算这些实例,创建另一个数组(z)并用z替换这些元素中的x:
x > y
# = [[False,False,True,True],# [False,True]]
在这种情况下,应该替换x(x [:,3:])的5个元素,因此我们创建一个(5,2)数组:
z = np.array([[20,21],[22,23],[24,25],[26,27],[28,29]])
我想要的结果是
x == np.array([[ 0,20,22,24,26,28],[ 8,21,23,25,27,29]])
最佳答案
一个几乎完全符合你想要的numpy函数是numpy.where:
x = np.arange(16).reshape((2,8))
y = np.array([[2],[13]])
z = np.arange(16,32).reshape((2,8))
numpy.where(~(x > y).any(axis=0),x,z)
结果:
array([[ 0,19,28,29,30,31]])
你要求的唯一区别是z必须可以播放到与x相同的形状.除非你绝对需要使用的z值与〜(x> y).any(axis = 0)中的True值一样多,我认为这是最好的方法.
但是,根据您的评论,您似乎确实需要使用如上所述的z值.听起来这个函数可以保证形状匹配,所以你可以这样做:
x[:,(x > y).any(axis=0)] = z.T
测试:
>>> z = np.arange(20,30).reshape((5,2))
>>> x[:,(x > y).any(axis=0)] = z.T
>>> x
array([[ 0,29]])
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。