diff column to wordcount CSV

[Ultimately_Untrue_Thought.git] / notes / memoir_wordcounts.py

diff --git a/notes/memoir_wordcounts.py b/notes/memoir_wordcounts.py
index 14aebc4..ac1aec4 100755 (executable)
--- a/notes/memoir_wordcounts.py
+++ b/notes/memoir_wordcounts.py
@@ -63,8 +63,11 @@ def normalize_dates(wordcounts):
 def write_csv(wordcounts):
     with open("memoir_wordcounts.csv", 'w') as f:
         writer = csv.writer(f)
+        previous = None
         for date, wordcount in wordcounts:
-            writer.writerow([date.strftime("%m/%d/%Y"), wordcount])
+            diff = wordcount - previous if previous is not None else ''
+            writer.writerow([date.strftime("%m/%d/%Y"), wordcount, diff])
+            previous = wordcount
 
 
 if __name__ in "__main__":