diff options
Diffstat (limited to 'python/interest_statement.py')
| -rw-r--r-- | python/interest_statement.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/python/interest_statement.py b/python/interest_statement.py index 716381ad..eddfd6f2 100644 --- a/python/interest_statement.py +++ b/python/interest_statement.py @@ -49,6 +49,16 @@ def f(df_balances, df_rates, broker, start_date, end_date): ) +def export_data(start, end): + dfs = {} + for cp in ("GS", "MS", "BAML_ISDA", "CITI", "CS", "BNP"): + dfs[cp] = f(df_balances, df_rates, cp, start, end) + df = pd.concat(dfs, names=["broker", "folder"]) + df = df[df.amount != 0.0] + df.amount *= -1.0 + return df + + if __name__ == "__main__": import argparse |
