我有这样一个数组:
array = ['A0','A1','A2','A3','A4','B0','B1','C0']
并希望获得一个数组,该数组对于后跟从0到2的数字的值为真。你知道吗
到目前为止,我是这样做的:
selection = np.where ((array == 'A0') | (array == 'A1') | (array == 'A2'), 1, 0)
但是,有没有更优雅的方法来实现这一点,比如使用一个正则表达式,比如:
selection = np.where (array == 'A[0-1]', 1, 0)
Tags:
如果选择使用熊猫:
我不认为numpy是你最好的解决方案。您可以使用内置的python工具(如
map
)来实现这一点。你知道吗如果它不比
A0
、A1
和A2
复杂,那么可以使用相关问题 更多 >
编程相关推荐