浮点对象的Python源代码中的注释确认:
比较几乎是一场噩梦
在将浮点数与整数进行比较时尤其如此,因为与浮点数不同,Python中的整数可以任意大,并且总是精确的。尝试将整数强制转换为浮点数可能会失去精度,并使比较不准确。尝试将浮点数转换为整数也不会起作用,因为任何小数部分都会丢失。
为了解决这个问题,Python执行了一系列检查,如果其中一项检查成功,则返回结果。它比较两个值的符号,然后比较整数是否“太大”而不能成为浮点数,然后将浮点的指数与整数长度进行比较。如果所有这些检查均失败,则有必要构造两个新的Python对象进行比较以获得结果。
将浮点数v
与整数/长整数进行比较时w
,最坏的情况是:
v
并且w
具有相同的符号(正号或负号),
- 该整数的
w
位数很少,可以保存为该size_t
类型(通常为32或64位),
- 整数
w
至少有49位,
- float的指数与中的
v
位数相同w
。
这正是我们对问题中的值所拥有的:
>>> import math
>>> math.frexp(562949953420000.7) # gives the float's (significand, exponent) pair
(0.9999999999976706, 49)
>>> (562949953421000).bit_length()
49
我们看到49既是浮点数的指数,也是整数的位数。这两个数字都是正数,因此符合上述四个条件。
选择一个较大或较小的值可以更改整数的位数或指数的值,因此Python能够确定比较结果,而无需执行昂贵的最终检查。
这特定于该语言的CPython实现。
比较比较详细
该float_richcompare
函数处理两个值v
和之间的比较w
。
下面是该功能执行的检查的分步说明。当试图了解函数的功能时,Python来源中的注释实际上非常有帮助,因此我将其放在相关的地方。我还将这些检查总结在答案底部的列表中。
其主要思想是映射Python对象v
和w
两个相应的C双打,i
并且j
,然后可以很容易地比较以得到正确的结果。Python 2和Python 3都使用相同的想法进行操作(前者只是分别处理int
和long
键入)。
首先要做的是检查v
是否绝对是Python float并将其映射到C double i
。接下来,该函数查看是否w
也是float并将其映射到C double j
。这是该功能的最佳情况,因为可以跳过所有其他检查。该功能还检查是否v
是inf
或nan
:
static PyObject*
float_richcompare(PyObject *v, PyObject *w, int op)
{
double i, j;
int r = 0;
assert(PyFloat_Check(v));
i = PyFloat_AS_DOUBLE(v);
if (PyFloat_Check(w))
j = PyFloat_AS_DOUBLE(w);
else if (!Py_IS_FINITE(i)) {
if (PyLong_Check(w))
j = 0.0;
else
goto Unimplemented;
}
现在我们知道,如果w
未通过这些检查,则不是Python浮点数。现在,该函数检查它是否为Python整数。在这种情况下,最简单的测试是提取v
和的符号w
(0
如果为零,-1
则返回,如果为负,1
则为正)。如果符号不同,则这是返回比较结果所需的全部信息:
else if (PyLong_Check(w)) {
int vsign = i == 0.0 ? 0 : i < 0.0 ? -1 : 1;
int wsign = _PyLong_Sign(w);
size_t nbits;
int exponent;
if (vsign != wsign) {
/* Magnitudes are irrelevant -- the signs alone
* determine the outcome.
*/
i = (double)vsign;
j = (double)wsign;
goto Compare;
}
}
如果此检查失败,则v
和w
具有相同的符号。
下一个检查将计算整数中的位数w
。如果它有太多位,那么就不可能将其保存为浮点数,因此其大小必须大于浮点数v
:
nbits = _PyLong_NumBits(w);
if (nbits == (size_t)-1 && PyErr_Occurred()) {
/* This long is so large that size_t isn't big enough
* to hold the # of bits. Replace with little doubles
* that give the same outcome -- w is so large that
* its magnitude must exceed the magnitude of any
* finite float.
*/
PyErr_Clear();
i = (double)vsign;
assert(wsign != 0);
j = wsign * 2.0;
goto Compare;
}
另一方面,如果整数w
的位数为48个或更少,则可以安全地将C翻倍j
并进行比较:
if (nbits <= 48) {
j = PyLong_AsDouble(w);
/* It's impossible that <= 48 bits overflowed. */
assert(j != -1.0 || ! PyErr_Occurred());
goto Compare;
}
从这一点开始,我们知道它w
具有49位或更多位。将其w
视为正整数会很方便,因此请根据需要更改符号和比较运算符:
if (nbits <= 48) {
/* "Multiply both sides" by -1; this also swaps the
* comparator.
*/
i = -i;
op = _Py_SwappedOp[op];
}
现在,该函数查看浮点数的指数。回想一下,可以将浮点数写为有效位数* 2 指数(忽略符号),并且有效位数表示0.5到1之间的数字:
(void) frexp(i, &exponent);
if (exponent < 0 || (size_t)exponent < nbits) {
i = 1.0;
j = 2.0;
goto Compare;
}
这检查了两件事。如果指数小于0,则浮点数小于1(因此,其大小小于任何整数)。或者,如果指数小于in的位数,w
则v < |w|
由于有效* 2 指数小于2 nbits,我们可以得到。
如果这两项检查均未通过,该函数将查看该指数是否大于中的位数w
。这表明有效数* 2 指数大于2 nbit,因此v > |w|
:
if ((size_t)exponent > nbits) {
i = 2.0;
j = 1.0;
goto Compare;
}
如果此检查未成功,我们将知道float的指数v
与整数中的位数相同w
。
现在可以比较两个值的唯一方法是从v
和构造两个新的Python整数w
。这个想法是丢弃的小数部分v
,将整数部分加倍,然后再加一个。w
也会加倍,并且可以将这两个新的Python对象进行比较以提供正确的返回值。使用具有较小值的示例,4.65 < 4
将由比较确定(2*4)+1 == 9 < 8 == (2*4)
(返回false)。
{
double fracpart;
double intpart;
PyObject *result = NULL;
PyObject *one = NULL;
PyObject *vv = NULL;
PyObject *ww = w;
// snip
fracpart = modf(i, &intpart); // split i (the double that v mapped to)
vv = PyLong_FromDouble(intpart);
// snip
if (fracpart != 0.0) {
/* Shift left, and or a 1 bit into vv
* to represent the lost fraction.
*/
PyObject *temp;
one = PyLong_FromLong(1);
temp = PyNumber_Lshift(ww, one); // left-shift doubles an integer
ww = temp;
temp = PyNumber_Lshift(vv, one);
vv = temp;
temp = PyNumber_Or(vv, one); // a doubled integer is even, so this adds 1
vv = temp;
}
// snip
}
}
为简洁起见,我省略了Python创建这些新对象时必须进行的其他错误检查和垃圾跟踪。不用说,这增加了额外的开销,并解释了为什么问题中突出显示的值比其他值慢得多。
这是比较功能执行的检查的摘要。
让它v
成为一个浮点数并将其转换为C的double。现在,如果w
也是浮点数:
如果w
为整数:
提取的迹象v
和w
。如果它们不同,那么我们知道v
并且w
不同,那是更大的价值。
(符号相同。)检查是否w
有太多位不能浮空(大于size_t
)。如果是这样,w
则其幅度大于v
。
检查是否w
有48位或更少的位。如果是这样,可以安全地将其强制转换为C double而不损失其精度,并与进行比较v
。
(w
具有超过48位。我们现在将w
其视作已适当更改比较操作的正整数。)
考虑浮点数的指数v
。如果指数为负,则v
小于1
且因此小于任何正整数。否则,如果指数小于的位数,w
则它必须小于w
。
如果的指数v
大于中的位数,w
则v
大于w
。
(指数与中的位数相同w
。)
最后检查。拆分v
成它的整数和小数部分。将整数部分加倍并加1以补偿小数部分。现在将整数倍w
。比较这两个新的整数以获得结果。