子集numpy数组基于另一个数组

Question

我有2个numpy数组：

arr_a = array(['1m_nd', '2m_nd', '1m_4wk'],
      dtype='<U15')

arr_b = array([0, 1, 1])

我想根据arr_a从arr_b中选择元素。我这样做： arr_a[arr_b]，但我得到了这个结果：

array(['1m_nd', '2m_nd', '2m_nd'],
      dtype='<U15')

而不是：

array(['2m_nd', '1m_4wk'],
      dtype='<U15')

我该如何解决这个问题？

Answer 1

你需要传递一个布尔数组，例如：

>>> arr_a[arr_b>0]

array(['2m_nd', '1m_4wk'], 
      dtype='<U15')

Answer 2

给定arr_a和arr_b，运行以下内容将为arr_b中值为1 => True和0 => False的每个元素提供布尔数组。相应地，使用arr_a中的索引值检查布尔值。以下是您需要的代码行。

>>> arr_a[arr_b == 1]
array([u'2m_nd', u'1m_4wk'],
      dtype='<U15')