python - Numpy 将 bool 数组的字符串表示形式转换为 bool 数组

Question

是否有一种原生的 numpy 方法来转换 bool 值的字符串表示数组，例如:

['True','False','True','False']

我可以使用一个实际的 bool 数组来屏蔽/索引？我可以做一个 for 循环遍历并重建数组，但对于大型数组来说，这很慢。

Answer 1

无论 dtype 是字符串还是 object，您都应该能够进行 bool 比较，IIUC:

>>> a = np.array(['True', 'False', 'True', 'False'])
>>> a
array(['True', 'False', 'True', 'False'], 
      dtype='|S5')
>>> a == "True"
array([ True, False,  True, False], dtype=bool)

或

>>> a = np.array(['True', 'False', 'True', 'False'], dtype=object)
>>> a
array(['True', 'False', 'True', 'False'], dtype=object)
>>> a == "True"
array([ True, False,  True, False], dtype=bool)