Mercurial Hosting > traffic-intelligence
comparison trafficintelligence/storage.py @ 1131:3972d85e3b6c
work on loading data for sumo
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 20 Feb 2020 23:56:56 -0500 |
parents | e53c6e87bb3f |
children | c4d9c270f999 |
comparison
equal
deleted
inserted
replaced
1130:18e2c8cb9b09 | 1131:3972d85e3b6c |
---|---|
942 connection.close() | 942 connection.close() |
943 | 943 |
944 ######################### | 944 ######################### |
945 # txt files | 945 # txt files |
946 ######################### | 946 ######################### |
947 | |
948 def loadCSVs(filenames, **kwargs): | |
949 '''Loads all the data from the filenames (eg from glob) and returns a concatenated dataframe''' | |
950 data = read_csv(filenames[0], **kwargs) | |
951 for f in filenames[1:]: | |
952 data = data.append(read_csv(filenames[0], **kwargs)) | |
953 return data | |
947 | 954 |
948 def saveList(filename, l): | 955 def saveList(filename, l): |
949 f = utils.openCheck(filename, 'w') | 956 f = utils.openCheck(filename, 'w') |
950 for x in l: | 957 for x in l: |
951 f.write('{}\n'.format(x)) | 958 f.write('{}\n'.format(x)) |