python - 测试 Python 中获取输入的不同方法的性能

Question

考虑 Python 3 的一行输入，如下所示:

1 4 99 8 7

代码需要将这些输入视为数字，因此设计了以下不同的方式来从用户处获取此或类似的输入:

方法一:

inputLst = [int(i) for i in input().split()]

方法二:

inputLst = map(int, input().split())

当用户在测试过程中必须与这样的代码交互时，如何使用两种不同的方法对代码进行准确的性能检查？有什么有效的办法吗？

Answer 1

输入时间是恒定的。如果将其替换为常量字符串，则不会改变速度差异。

如果您使用 timeit 模块，请在语句中将 input() 替换为常量，或者在设置中声明该函数 导入时间 随机导入

tests = []
random_input = ' '.join(str(random.randint(1000, 9999)) for _ in range(1000))
setup = 'def input():\n    return ' + repr(random_input)

tests.append(timeit.timeit('[int(i) for i in input().split()]', setup, number=10000))
tests.append(timeit.timeit('list(map(int, input().split()))', setup, number=10000))
for i, time in enumerate(tests, 1):
    print('Test #{} took {:.3f} seconds to run 10000 trials.'.format(i, time))

输出是(Python2):

Test #1 took 2.477 seconds to run 10000 trials.
Test #2 took 2.079 seconds to run 10000 trials.

(Python3):

Test #1 took 1.448 seconds to run 10000 trials.
Test #2 took 1.152 seconds to run 10000 trials.

所以 map() 在两个版本中确实更快。