n26_plot_csv.py

import argparse
import csv
from datetime import datetime
from matplotlib import pyplot
import numpy as np

parser = argparse.ArgumentParser(description='Plot N26 CSV')
parser.add_argument('filepath', type=str, help='path to csv file downloaded from N26')

args = parser.parse_args()

balances = []
dates = []
transactions = []
categories = set()
with open(args.filepath, 'r') as f:
    reader = csv.reader(f, delimiter=',', quotechar='"')
    # skip header
    next(reader)
    current_balance = 0
    for i, row in enumerate(reader):
        date, payee, account_number, transaction_type, payment_reference, category, amount_eur, amount_foreign, type_foreign, exchange_rate = row
        date = datetime.strptime(date, '%Y-%m-%d').date()
        amount_eur = float(amount_eur)

        current_balance += amount_eur
        categories.add(category)
        balances.append(current_balance)
        transactions.append(amount_eur)
        dates.append(date)

print(categories)
pyplot.bar(dates, [max(0, t) for t in transactions], color='g', linewidth=0)
pyplot.bar(dates, [min(0, t) for t in transactions], color='r', linewidth=0)
pyplot.plot(dates, balances, marker='x', linestyle='-')
pyplot.show()