Numpy dtype = int

问题描述 投票:4回答:1

在下面的代码中。我得到了x1的预期结果

import numpy as np 
x1 = np.arange(0.5, 10.4, 0.8)
print(x1)
[ 0.5  1.3  2.1  2.9  3.7  4.5  5.3  6.1  6.9  7.7  8.5  9.3 10.1]

但是在下面的代码中,当我设置dtype = int时为什么x2的结果不是[ 0 1 2 2 3 4 5 6 6 7 8 9 10]而是我得到x2的值为[ 0 1 2 3 4 5 6 7 8 9 10 11 12],其中最后一个值12超过了10.4的结束值。请澄清我的概念。

import numpy as np 
x2 = np.arange(0.5, 10.4, 0.8, dtype=int)
print(x2)
[ 0  1  2  3  4  5  6  7  8  9 10 11 12]
python python-3.x numpy numpy-ndarray numerical-computing
1个回答
4
投票

根据文件:https://docs.scipy.org/doc/numpy1.15.0/reference/generated/numpy.arange.html

停止:数字间隔结束。间隔不包括此值,除非在某些情况下,step不是整数,浮点舍入会影响out的长度。

arange:ndarray均匀间隔值的数组。 对于浮点参数,结果的长度为ceil((stop - start)/ step)。由于浮点溢出,此规则可能导致out的最后一个元素大于stop。

所以这里最后一个元素是。

In [33]: np.ceil((10.4-0.5)/0.8)                                                
Out[33]: 13.0

因此,在np.arange(0.5, 10.4, 0.8, dtype=int)的情况下,我们看到过冲为12,因为在上述情况下为stop=13,默认值为0,

因此我们观察到的输出是。

In [35]: np.arange(0.5, 10.4, 0.8, dtype=int)                                   
Out[35]: array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12])

因此,生成整数范围的更好方法是使用整数参数,如下所示:

In [25]: np.arange(0, 11, 1)                                                    
Out[25]: array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10])
© www.soinside.com 2019 - 2024. All rights reserved.