summaryrefslogtreecommitdiff
path: root/Chapter14ProjRemoveHeadersCsv.py
diff options
context:
space:
mode:
Diffstat (limited to 'Chapter14ProjRemoveHeadersCsv.py')
-rw-r--r--Chapter14ProjRemoveHeadersCsv.py33
1 files changed, 33 insertions, 0 deletions
diff --git a/Chapter14ProjRemoveHeadersCsv.py b/Chapter14ProjRemoveHeadersCsv.py
new file mode 100644
index 0000000..cf3d205
--- /dev/null
+++ b/Chapter14ProjRemoveHeadersCsv.py
@@ -0,0 +1,33 @@
+#! python3
+
+# Chapter 14 Project Removing the Header from CSV Files
+# Removes the header from all CSV files in the current working
+# directory.
+
+import os
+import csv
+
+os.makedirs('headerRemoved', exist_ok=True)
+
+for csvFilename in os.listdir('.'):
+ if not csvFilename.endswith('.csv'):
+ continue
+
+ print('Removing header from ' + csvFilename + '...')
+
+ csvRows = []
+ csvFileObj = open(csvFilename)
+ readerObj = csv.reader(csvFileObj)
+ for row in readerObj:
+ if readerObj.line_num == 1:
+ continue
+ csvRows.append(row)
+ csvFileObj.close()
+
+ csvFileObj = open(os.path.join('headerRemoved', csvFilename), 'w', newline='')
+ csvWriter = csv.writer(csvFileObj)
+ for row in csvRows:
+ csvWriter.writerow(row)
+ csvFileObj.close()
+
+