如何解决高级布尔索引
我想通过掩码选择值并使用掩码数组更改值。
代码:
import numpy as np
a = np.zeros((2,2),dtype=(np.uint8,3))
x = np.arange(4,dtype=int).reshape((2,2))
mask = np.logical_and(a1 < 3,a1 > 0)
a[mask] = (1,x[mask],2)
我想要结果:
a[mask]
>> [[1,1,2],[1,2,2]]
但我收到错误:
ValueError: setting an array element with a sequence.
如果尝试做 a[mask] = (1,2)
之类的事情
数组将是
[[[0,0],2]],[[1,[0,0]]]
但我需要使用来自 x 的值。 让它看起来像
[[[0,3]],3],0]]]
我该怎么做?
解决方法
可以分两步完成。
import numpy as np
a = np.zeros((2,2),dtype=(np.uint8,3))
x = np.arange(4,dtype=int).reshape((2,2))
a1 = x # Create an a1 for the mask
mask = np.logical_and(a1 < 3,a1 > 0)
a[mask] = (1,2) # Set the outer columns
a[mask,1] = x[mask] # Set the column 1
print( a )
# [[[0 0 0]
# [1 1 2]]
# [[1 2 2]
# [0 0 0]]]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。