From b87ec4e7758a26e91e1a4e5f78df91fafd196f8d Mon Sep 17 00:00:00 2001 From: mzuber Date: Thu, 5 Dec 2024 10:18:19 +0000 Subject: [PATCH] Use for comprehension to speed up python code --- loops/py/code.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/loops/py/code.py b/loops/py/code.py index 119c0060..c707b4da 100644 --- a/loops/py/code.py +++ b/loops/py/code.py @@ -1,15 +1,11 @@ -import sys import random +import sys def main(): u = int(sys.argv[1]) # Get an input number from the command line r = random.randint(0, 10000) # Get a random number 0 <= r < 10k - a = [0] * 10000 # Array of 10k elements initialized to 0 - for i in range(10000): # 10k outer loop iterations - for j in range(100000): # 100k inner loop iterations, per outer loop iteration - a[i] += j % u # Simple sum - a[i] += r # Add a random value to each element in array + a = [sum([j % u for j in range(100_000)]) + r for _ in range(10_000)] print(a[r]) # Print out a single element from the array