Search code examples
pythonpandassum

Calculate a sum on a range in column in pandas

I would like to calculate a sum on a range in column in panda dataframe in python.

Panda DataFrame:

acc     accname     amount
1010    turnover    10000
2000    salaries    -4000
3000    rent        -1500
5000    assets      15000
6000    liabilities -15000

I would like to calculate the result. 10000-4000-1500 =4500 or sum(1010:3000). And I would like to define the value as a variable called result.

Solution

  • You can use pandas.DataFrame.set_index and pandas.DataFrame.loc :

    result = df.set_index('acc').loc[1010:3000, 'amount'].sum()

    # Output :

    print(result)
4500