Helper to filter CSV columns you need

csv_columns_subset.py

#!/usr/bin/python3
import csv
import sys

csv_reader = csv.DictReader(sys.stdin)
csv_writer = csv.DictWriter(sys.stdout, fieldnames=[h for h in csv_reader.fieldnames if h in sys.argv[1:]])
csv_writer.writeheader()
for row in csv_reader:
    csv_writer.writerow({f: row[f] for f in sys.argv[1:]})