blob: 7dbc155c79e903fe9857eb06755d08a307ffef85 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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])
|