timeit库可以用于测试代码的性能
1.测试单行代码的运行时间:
import timeit
code = "for i in range(100): pass"
time = timeit.timeit(code, number=10000)
print(time)
2.测试一个函数的运行时间:
import timeit
def my_function():
for i in range(100): pass
time = timeit.timeit(my_function, number=10000)
print(time)
3.测试两个函数的运行时间:
import timeit
def my_function1():
for i in range(100): pass
def my_function2():
for i in range(100): pass
time = timeit.timeit((my_function1, my_function2), number=10000)
print(time)
4.测试一行代码的平均运行时间:
import timeit
code = "for i in range(100): pass"
time = timeit.timeit(code, number=10000)
average_time = time / 10000
print(average_time)
5.测试一个函数的平均运行时间:
import timeit
def my_function():
for i in range(100): pass
time = timeit.timeit(my_function, number=10000)
average_time = time / 10000
print(average_time)
6.测试两个函数的平均运行时间:
import timeit
def my_function1():
for i in range(100): pass
def my_function2():
for i in range(100): pass
time = timeit.timeit((my_function1, my_function2), number=10000)
average_time = time / 10000
print(average_time)
7.测试多行代码的运行时间:
import timeit
code = """
for i in range(100):
x = i * i
"""
time = timeit.timeit(code, number=10000)
print(time)
8.测试一个函数的多个参数的运行时间:
import timeit
def my_function(x, y):
for i in range(x * y): pass
time = timeit.timeit(lambda: my_function(10, 20), number=10000)
print(time)
9.测试一个函数的多个参数组合的运行时间:
import timeit
def my_function(x, y):
for i in range(x * y): pass
time = timeit.timeit(lambda: my_function(10, 20), number=10000)
print(time)
10.测试一个函数的不同输入参数的运行时间:
import timeit
def my_function(x):
for i in range(x): pass
for i in range(1, 6):
time = timeit.timeit(lambda: my_function(i * 10), number=10000)
print(f"Time for input {i * 10}: {time}")
直接计数:
import time
start_time = time.perf_counter()
# 这里是要测试运行时间的代码段
for i in range(1000000):
pass
end_time = time.perf_counter()
run_time = end_time - start_time
print("代码运行时间:", run_time)