Add more scripts

1 files changed, 26 insertions, 0 deletions

diff --git a/scripts/scale.py b/scripts/scale.py
new file mode 100755
index 0000000..7dbc155
--- /dev/null
+++ b/scripts/scale.py
@@ -0,0 +1,26 @@
+#!/usr/bin/env python3
+
+import csv
+import sys
+import random
+
+def main(filename, output_file):
+    with open(filename, "r") as f:
+        reader = list(csv.reader(f))
+    newreader = []
+    for row in reader:
+        try:
+            if float(row[4]) > 900:
+                row[4] = "900"
+            if float(row[3]) > 900:
+                row[3] = "900"
+            if random.random() < 0.25:
+                newreader.append(row)
+        except:
+            newreader.append(row)
+    with open(output_file, "w") as f:
+        writer = csv.writer(f)
+        writer.writerows(newreader)
+
+if __name__ == "__main__":
+    main(sys.argv[1], sys.argv[2])