python - 如何在 Python 3 中启用 xrange 以实现可移植性？

Question

我写了一个脚本，我想为 Python 2 和 Python 3 启用它。

从 __future__ 导入 division 和 print_function 后，我唯一担心的是我的 range 返回了一个完整的数组在 Python 2 中，浪费时间和内存。

作为解决方法，我在脚本开头添加了以下 3 行:

if sys.version_info[0] == 3:
    def xrange(i):
        return range(i)

然后，我只在我的代码中使用了xrange。

有没有比我的解决方法更优雅的方法？

Answer 1

你可以稍微简化一下:

if sys.version_info[0] == 3:
    xrange = range

我会反过来做:

if sys.version_info[0] == 2:
    range = xrange

如果您想放弃对 Python 2.x 的支持，只需删除这两行即可，而无需检查所有代码。

但是，我强烈建议考虑 six library .它是实现 Python 2 和 3 兼容性的事实标准。

from six.moves import range