Group budget balancing

gbudget.py

import pandas as pd

# Choose the file to balance
csvFile = 'example'

# Clear the log
open('log', 'w').close()
# Open the csv file
df = pd.read_csv('data/{0}.csv'.format(csvFile))
# Obtain sum of amount paid
sums = df.groupby('person').sum()
# Compute the mean total amount
mean = sums['amount'].mean()
# Compute the differences
differences = sums - mean

# Now let's find the steps to balance everything out

while sum(abs(differences['amount'])) != 0:
    receiver = differences['amount'].argmax()
    giver = differences['amount'].argmin()
    amount = min(differences.loc[receiver]['amount'],
                 abs(differences.loc[giver]['amount']))
    # Record the transaction in a file
    with open('log', 'a') as log:
        log.write(str(giver) +' => '+str(amount)+' => '+str(receiver)+'\n')
    # Modify the data
    differences.loc[receiver] -= amount
    differences.loc[giver] += amount