Creating a Python decorator to measure function execution time

Last update on December 21 2024 07:28:02 (UTC/GMT +8 hours)

Write a Python program to create a decorator function to measure the execution time of a function.

Sample Solution:

Python Code:

import time

def measure_execution_time(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        execution_time = end_time - start_time
        print(f"Function {func.__name__} took {execution_time:.4f} seconds to execute")
        return result
    return wrapper

# Example usage
@measure_execution_time
def calculate_multiply(numbers):
    tot = 1
    for x in numbers:
        tot *= x
    return tot

# Call the decorated function
result = calculate_multiply([1, 2, 3, 4, 5])
print("Result:", result)

Sample Output:

Function calculate_multiply took 0.0000 seconds to execute
Result: 120

Explanation:

In the above exercise -

• The execution_time variable is calculated by subtracting start_time from end_time, representing the elapsed time.
• The execution time is printed to the console, indicating which function was measured and how long it took. This is done using the format string f"Function {func.__name__} took {execution_time:.4f} seconds to execute". The .4f format specifier ensures four decimal places for execution time.
• Finally, the original function result is returned.

The @execution_time syntax is used to apply the execution_time decorator to the calculate_multiply function. This indicates that it should be decorated with the functionality provided by the execution_time function.

Flowchart:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Creating a Python decorator to log function arguments and return value.
Next: Creating a Python decorator to convert function return value.