Revize cecb4388
Přidáno uživatelem Petr Hlaváč před asi 4 roky(ů)
python-module/Utilities/CSV/CSVutils.py | ||
---|---|---|
4 | 4 |
|
5 | 5 |
def get_unique_names_from_file(filename, column_number): |
6 | 6 |
""" |
7 |
|
|
7 |
Extract set of unique names from file |
|
8 | 8 |
Args: |
9 |
filename: |
|
10 |
column_number: |
|
9 |
filename: path to processed file
|
|
10 |
column_number: unique names are expected in csv file on column_number
|
|
11 | 11 |
|
12 | 12 |
Returns: |
13 |
|
|
13 |
set of unique names |
|
14 | 14 |
""" |
15 |
f = open(filename, "r") |
|
16 |
|
|
17 | 15 |
# create set of unique names |
18 | 16 |
name_set = set() |
19 | 17 |
|
20 |
# go through every line of line |
|
21 |
for x in f: |
|
22 |
# split by csv splitter ; |
|
23 |
array = x.split(";") |
|
24 |
# add string from chosen column to set |
|
25 |
name_set.add(array[column_number]) |
|
26 |
|
|
27 |
f.close() |
|
18 |
with open(filename, "r") as file: |
|
19 |
# go through every line of line |
|
20 |
for x in file: |
|
21 |
# split by csv splitter ; |
|
22 |
array = x.split(";") |
|
23 |
# add string from chosen column to set |
|
24 |
name_set.add(array[column_number]) |
|
28 | 25 |
|
29 | 26 |
return name_set |
30 | 27 |
|
31 | 28 |
|
32 | 29 |
def export_data_to_csv(filename, data_dict): |
30 |
""" |
|
31 |
Takes data_dict and export it into a csv file |
|
32 |
Args: |
|
33 |
filename: name of exported file |
|
34 |
data_dict: dictionary containing data from DatasetProcessor |
|
35 |
""" |
|
33 | 36 |
with open(PROCESSED_DATA_PATH + filename[12:], "w+") as file: |
34 | 37 |
|
35 | 38 |
for date in data_dict: |
Také k dispozici: Unified diff
Fixed issues found in code documentation