SIP contribution calculation script

calc.py

def main():
    sip = 5000
    years = 5
    num_of_months = 12*years
    total_corpus = 0

    average_annual_return = 6.0

    for month in range(num_of_months):
        if month % 6 == 0 and month != 0:
            sip *= 1.1

        total_corpus += sip

        print(f'Month: {month} : {sip:.2f}, {total_corpus:.2f}')

    expected_return = years * total_corpus * average_annual_return / 100

    print(f'Total Investment: {total_corpus:.2f}, Expected Return: {expected_return:.2f}, Total: {(total_corpus + expected_return): .2f}')


if __name__ == "__main__":
    main()