将 numpy 布尔数组转换为具有不同值的 int 数组

Question

我已经阅读了大部分 How to convert a boolean array to an int array ，但我仍然不知道如何（最有效地）将 numpy bool 数组转换为 int 数组，但具有不同的值。例如，我有：

>>> k=np.array([True, False, False, True, False])
>>> print k
[ True False False  True False]

我希望将其转换为数组，其中 True 表示 2，False 表示 5。

当然，我总是可以建立一个线性方程：

>>> print 5-k*3
[2 5 5 2 5]

...虽然是矢量化的，但不必要地同时使用了加法（减法）和乘法；此外，如果我想要相反的值（True 为 5，False 为 2），我基本上必须使用（并重新计算）不同的方程：

>>> print 2+k*3
[5 2 2 5 2]

... 这对我来说有点可读性问题。

本质上，这只是一个选择/映射操作——但我想在numpy 域中完成。我怎么能这样做？

Answer 1

似乎numpy.where 正是你想要的：

>>> import numpy as np
>>> k = np.array([True, False, False, True, False])
>>> np.where(k, 2, 5)
array([2, 5, 5, 2, 5])

Answer 2

好吧，看来我必须创建一个包含两个不同值的 np.array（不是 Python 列表！）：

>>> z=np.array([2,5])
>>> print z
[2 5]

...然后我可以简单地将布尔数组 (k) 转换为 int，并将其用作“不同值数组”(z) 中的选择索引：

>>> print z[k.astype(int)]
[5 2 2 5 2]

将 [2,5] 更改为 [5,2] 也很简单，这样很好。

我只是不确定这是否是正确的做法（也就是说，可能存在类似（伪代码）k.asdistinctvalues([2,5]) 或类似的东西？）