如何在python numpy ndarray中添加一个点 - 数据类型问题

Question

我的NumPy ndarray看起来像是：

[[ 0 0 0 1 0]
 [ 0 0 0 0 1]]

但我想将其处理为以下形式：

[[   0.    0.    0.    1.    0.]
 [   0.    0.    0.    0.    1.]]

我将如何实现这一目标？

Answer 1

在我看来，你有一个整数类型的数组。您可能希望转换为float数组：

array_float = array_int.astype(float)

e.g：

>>> ones_i = np.ones(10, dtype=int)
>>> print ones_i
[1 1 1 1 1 1 1 1 1 1]
>>> ones_f = ones_i.astype(float)
>>> print ones_f
[ 1.  1.  1.  1.  1.  1.  1.  1.  1.  1.]

话虽如此，我认为值得问为什么要处理数组的字符串表示。很有可能是实现目标的更好方法。