时间戳

首页 > 解决方案 > Numpy floor float 值转换为 int

python - Numpy floor float 值转换为 int

问题描述

我有浮点数组，我想将它们设为最接近的整数，所以我可以将它们用作索引。

例如：

In [2]: import numpy as np

In [3]: arr = np.random.rand(1, 10) * 10

In [4]: arr
Out[4]:
array([[4.97896461, 0.21473121, 0.13323678, 3.40534157, 5.08995577,
        6.7924586 , 1.82584208, 6.73890807, 2.45590354, 9.85600841]])

In [5]: arr = np.floor(arr)

In [6]: arr
Out[6]: array([[4., 0., 0., 3., 5., 6., 1., 6., 2., 9.]])

In [7]: arr.dtype
Out[7]: dtype('float64')

地板后它们仍然是浮点数，有没有办法自动将它们转换为整数？

标签： pythonnumpymathfloor

解决方案

我用@DanielF 解释编辑答案：“floor 不会转换为整数，它只是给出整数值的浮点数，所以你仍然需要一个 astype 来更改为 int”检查此代码以了解解决方案：

import numpy as np
arr = np.random.rand(1, 10) * 10
print(arr)
arr = np.floor(arr).astype(int)
print(arr)

OUTPUT:
[[2.76753828 8.84095843 2.5537759  5.65017407 7.77493733 6.47403036
  7.72582766 5.03525625 9.75819442 9.10578944]]
[[2 8 2 5 7 6 7 5 9 9]]

推荐阅读