add writing column names in logfile

2022-02-21 17:27:08 -06:00
parent 5c297fa4d1
commit 7d8f7855d5
1 changed files with 7 additions and 0 deletions
--- a/tools/python/pizza/log.py
+++ b/tools/python/pizza/log.py
@ -171,6 +171,13 @@ class log:
      colmap = range(self.nvec)

    f = open(filename,"w")
+
+    # write col names from dict in the right order
+    colnames = [k for j in colmap for k,v in self.ptr.items() if v == j]
+    for j in range(len(colnames)):
+        print(colnames[j], file=f, end=" ")
+    print("\n", file=f, end="")
+
    # write data
    for i in range(self.nlen):
      for j in range(len(colmap)):