Skip to content

Commit 7388971

Browse files
authored
Update examples.py
1 parent 8fa134e commit 7388971

1 file changed

Lines changed: 26 additions & 0 deletions

File tree

optimize-python-code/examples.py

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,27 @@
1+
# before
2+
import time
3+
4+
def square_numbers_loop(numbers):
5+
result = []
6+
for num in numbers:
7+
result.append(num ** 2)
8+
return result
9+
10+
# Let's test this with 100,000 numbers to see the performance
11+
test_numbers = list(range(1000000))
12+
13+
start_time = time.time()
14+
squared_loop = square_numbers_loop(test_numbers)
15+
loop_time = time.time() - start_time
16+
print(f"Loop time: {loop_time:.4f} seconds")
17+
18+
# after
19+
def square_numbers_comprehension(numbers):
20+
return [num ** 2 for num in numbers] # Create the entire list in one line
21+
22+
start_time = time.time()
23+
squared_comprehension = square_numbers_comprehension(test_numbers)
24+
comprehension_time = time.time() - start_time
25+
print(f"Comprehension time: {comprehension_time:.4f} seconds")
26+
print(f"Improvement: {loop_time / comprehension_time:.2f}x faster")
127

0 commit comments

Comments
 (0)