|
11 | 11 | def write_file(args): |
12 | 12 | file_name, rows, curr_time, delta_time, curr_vals, delta_val = args |
13 | 13 | fieldnames = ['timestamp'] |
14 | | - fieldnames.extend(['C{0:d}'.format(i + 1) for i in range(len(curr_vals))]) |
| 14 | + fieldnames.extend([f'C{i + 1:d}' for i in range(len(curr_vals))]) |
15 | 15 | with open(file_name, 'w', newline='') as csv_file: |
16 | 16 | writer = csv.DictWriter(csv_file, fieldnames=fieldnames) |
17 | 17 | writer.writeheader() |
18 | | - values = [] |
19 | | - for _ in range(rows//2): |
20 | | - data = {'C{0:d}'.format(i + 1): curr_vals[i] |
21 | | - for i in range(len(curr_vals))} |
| 18 | + for _ in range(rows): |
| 19 | + data = {f'C{i + 1:d}': val for i, val in enumerate(curr_vals)} |
22 | 20 | data['timestamp'] = curr_time |
23 | 21 | writer.writerow(data) |
24 | | - values.append(data) |
25 | 22 | curr_time += delta_time |
26 | 23 | curr_vals = [x + random.uniform(-delta_val, delta_val) |
27 | 24 | for x in curr_vals] |
28 | | - while values: |
29 | | - data = values.pop() |
30 | | - data['timestamp'] = curr_time |
31 | | - writer.writerow(data) |
32 | | - curr_time += delta_time |
33 | 25 | return file_name |
34 | 26 |
|
35 | 27 |
|
|
0 commit comments