Stowage/cpython
2
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-10-31 13:41:24 +00:00
Code Issues Projects Releases Packages Wiki Activity

bpo-41902: Micro optimization for range.index if step is 1 (GH-22479)

Browse source
This commit is contained in:
Dong-hee Na 2020-10-21 11:29:56 +09:00 • committed by GitHub
parent 5f22741340
commit c0f22fb8b3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

16
Objects/rangeobject.c
View file

@ -582,13 +582,19 @@ range_index(rangeobject *r, PyObject *ob)
return NULL;
if (contains) {
PyObject *idx, *tmp = PyNumber_Subtract(ob, r->start);
if (tmp == NULL)
PyObject *idx = PyNumber_Subtract(ob, r->start);
if (idx == NULL) {
return NULL;
}
if (r->step == _PyLong_One) {
return idx;
}
/* idx = (ob - r.start) // r.step */
idx = PyNumber_FloorDivide(tmp, r->step);
Py_DECREF(tmp);
return idx;
PyObject *sidx = PyNumber_FloorDivide(idx, r->step);
Py_DECREF(idx);
return sidx;
}
/* object is not in the range */