我有以下 numpy 数组:
array(['NONRBP', 'RBP', 'NONRBP', 'RBP'], dtype=object)
我想做的是屏蔽上面的数组，给 RBP 赋值 1，给 NONRBP 赋值 0，结果是:
array([0, 1, 0, 1])
有什么方法可以做到？
请您参考如下方法:
对于更像 numpy 的解决方案，您可以这样做:
a = array(['NONRBP', 'RBP', 'NONRBP', 'RBP'], dtype=object)
mask = (a == 'RBP').astype(int)
您还可以通过以下方式进行操作:
a[:] = (a == 'RBP').astype(int)
