From 5003ac0f4fec51f00b334344ac30db15dc1db05c Mon Sep 17 00:00:00 2001 From: Conrad Date: Thu, 17 May 2018 20:16:39 +0200 Subject: [PATCH] Performance boost 3 sec gain! --- src/data_generators/randomi.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/src/data_generators/randomi.py b/src/data_generators/randomi.py index f306f78..4aa71ab 100644 --- a/src/data_generators/randomi.py +++ b/src/data_generators/randomi.py @@ -5,14 +5,13 @@ import time from datetime import date def main(i, a, b): - for x in range(0, int(i)): - content = generateFile(a, b) - writeFile(content, x) + for x in range(0, i): + writeFile(generateFile(a, b), x) def generateFile(rows, lenghtOfRows): content = [] - for y in range(0, int(rows)): - content.append(generateRow(int(lenghtOfRows))) + for y in range(0, rows): + content.append(generateRow(lenghtOfRows)) return content def generateRow(lenghtOfRows): @@ -27,9 +26,9 @@ def writeFile(content, x): file.write(content[w] + "\n") print("Hello World") -i = input("How many datapices would you like to generate?") -a = input("How many rows should each datapiece have?") -b = input("How long should each row be?") +i = int(input("How many datapices would you like to generate? ")) +a = int(input("How many rows should each datapiece have? ")) +b = int(input("How long should each row be? ")) start_time = time.time()