|
@@ -0,0 +1,29 @@
|
|
1
|
+from argparse import ArgumentParser
|
|
2
|
+from sys import stdout
|
|
3
|
+import pandas as p
|
|
4
|
+
|
|
5
|
+def main():
|
|
6
|
+ parser = ArgumentParser(description='Transform a "pickle" to a csv file')
|
|
7
|
+ parser.add_argument("-i", "--input", dest="input", help = "input pickle path", metavar="PATH", required = True)
|
|
8
|
+ parser.add_argument("-o", "--output", dest="output", help = "output csv path", metavar="PATH")
|
|
9
|
+ parser.add_argument("-r", "--row-names", dest = "rownames", help = "include row names in csv; if ommitted prints to stdout", action = "store_true")
|
|
10
|
+ parser.add_argument("-v", "--verbose", dest = "verbose", action ="store_true")
|
|
11
|
+ args = parser.parse_args()
|
|
12
|
+
|
|
13
|
+ if args.output is None:
|
|
14
|
+ args.output = stdout
|
|
15
|
+
|
|
16
|
+ if args.verbose:
|
|
17
|
+ print("Reading pickle")
|
|
18
|
+
|
|
19
|
+ ptc = p.read_pickle(args.input)
|
|
20
|
+
|
|
21
|
+ if args.verbose:
|
|
22
|
+ print(ptc.info())
|
|
23
|
+ print("Saving as csv")
|
|
24
|
+
|
|
25
|
+ ptc.to_csv(args.output, index = args.rownames)
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+if __name__ == "__main__":
|
|
29
|
+ main()
|