Skip to main content

Python Pandas: How to Sum Column Values Based on Condition(s)

Calculating conditional sums in a Pandas DataFrame is a frequent operation in data analysis. You might need to sum values in one column only for rows where another column meets specific criteria, or where multiple conditions across different columns are satisfied. Pandas provides powerful and concise ways to achieve this using boolean indexing and the DataFrame.query() method.

This guide explains how to sum values in a Pandas DataFrame column based on single or multiple conditions.

The Goal: Conditional Summation of a Column​

Given a Pandas DataFrame, we want to calculate the sum of values in a specific target column, but only for those rows that satisfy one or more conditions based on values in other (or the same) columns. For example, summing the 'SalesAmount' for all transactions in the 'North' region.

Example DataFrame​

import pandas as pd
import numpy as np # For potential NaN if needed

data = {
    'Region': ['North', 'South', 'North', 'West', 'South', 'North', 'East', 'West'],
    'Product_Category': ['Electronics', 'Books', 'Apparel', 'Electronics', 'Books', 'Apparel', 'Home', 'Books'],
    'Sales_Amount': [1500, 800, 1200, 2200, 600, 900, 1800, 750],
    'Units_Sold': [10, 50, 30, 15, 40, 25, 60, 45]
}
df = pd.DataFrame(data)
print("Original DataFrame:")
print(df)

Output:

We'll perform conditional sums on the 'Sales_Amount' column.

Sum Values Based on a SINGLE Condition​

  1. Create a boolean Series representing the condition.
  2. Use df.loc[condition_mask, 'column_to_sum'] to select the values in the target column where the condition is true.
  3. Call .sum() on the resulting Series.
import pandas as pd

df_example = pd.DataFrame({
    'Region': ['North', 'South', 'North', 'West'],
    'Sales_Amount': [1500, 800, 1200, 2200]
})

# Condition: Region is 'North'
condition_north = (df_example['Region'] == 'North')
print("Boolean mask for Region == 'North':")
print(condition_north)

# ✅ Select 'Sales_Amount' for rows where condition is True, then sum
sum_sales_north = df_example.loc[condition_north, 'Sales_Amount'].sum()

print(f"Total Sales Amount for 'North' region (using .loc): {sum_sales_north}")

Output:

Boolean mask for Region == 'North':
0     True
1    False
2     True
3    False
Name: Region, dtype: bool
Total Sales Amount for 'North' region (using .loc): 2700

Using Boolean Indexing (Shorter Syntax)​

You can achieve the same with a slightly more concise boolean indexing syntax if you are selecting a single column to sum.

import pandas as pd

df_example = pd.DataFrame({
    'Region': ['North', 'South', 'North', 'West'],
    'Sales_Amount': [1500, 800, 1200, 2200]
})

# ✅ Shorter syntax: df[condition]['column_to_sum'].sum()
sum_sales_north_short = df_example[df_example['Region'] == 'North']['Sales_Amount'].sum()

print(f"Total Sales Amount for 'North' region (shorter syntax): {sum_sales_north_short}")

Output:

Total Sales Amount for 'North' region (shorter syntax): 2700

While shorter, using .loc (as in 3.1) is often preferred for its explicitness and to avoid potential SettingWithCopyWarning in more complex scenarios if assignments were involved.

Using DataFrame.query()​

The query() method allows string-based condition specification.

import pandas as pd

df_example = pd.DataFrame({
    'Region': ['North', 'South', 'North', 'West'],
    'Sales_Amount': [1500, 800, 1200, 2200]
})

# ✅ Use query() to filter rows, then select column and sum
sum_sales_north_query = df_example.query("Region == 'North'")['Sales_Amount'].sum()

print(f"Total Sales Amount for 'North' region (using query()): {sum_sales_north_query}")

Output:

Total Sales Amount for 'North' region (using query()): 2700

Sum Values Based on MULTIPLE Conditions (AND Logic)​

All specified conditions must be true for a row's value to be included in the sum.

Using Boolean Indexing with &​

Combine individual boolean conditions using the logical AND operator &. Remember to wrap each condition in parentheses ().

import pandas as pd

df_example = pd.DataFrame({
    'Region': ['North', 'South', 'North', 'West', 'North'],
    'Product_Category': ['Electronics', 'Books', 'Apparel', 'Electronics', 'Electronics'],
    'Sales_Amount': [1500, 800, 1200, 2200, 1000]
})

# Condition: Region is 'North' AND Product_Category is 'Electronics'
cond_region_north = (df_example['Region'] == 'North')
cond_category_elec = (df_example['Product_Category'] == 'Electronics')

# ✅ Combine with &
sum_sales_north_elec = df_example.loc[cond_region_north & cond_category_elec, 'Sales_Amount'].sum()

print("Total Sales for 'North' AND 'Electronics':")
print(sum_sales_north_elec)

Output:

Total Sales for 'North' AND 'Electronics':
2500

Using DataFrame.query() with and​

import pandas as pd

df_example = pd.DataFrame({
    'Region': ['North', 'South', 'North', 'West', 'North'],
    'Product_Category': ['Electronics', 'Books', 'Apparel', 'Electronics', 'Electronics'],
    'Sales_Amount': [1500, 800, 1200, 2200, 1000]
})

sum_sales_north_elec_query = df_example.query(
    "Region == 'North' and Product_Category == 'Electronics'"
)['Sales_Amount'].sum()

print("Total Sales for 'North' AND 'Electronics' (query()):")
print(sum_sales_north_elec_query)

Output:

Total Sales for 'North' AND 'Electronics' (query()):
2500

Sum Values if AT LEAST ONE of Multiple Conditions is Met (OR Logic)​

At least one of the specified conditions must be true.

Using Boolean Indexing with |​

Combine individual boolean conditions using the logical OR operator |. Wrap each condition in parentheses.

import pandas as pd

df_example = pd.DataFrame({
    'Region': ['North', 'South', 'North', 'West', 'South'],
    'Product_Category': ['Electronics', 'Books', 'Apparel', 'Electronics', 'Books'],
    'Sales_Amount': [1500, 800, 1200, 2200, 600]
})

# Condition: Region is 'South' OR Product_Category is 'Electronics'
cond_region_south = (df_example['Region'] == 'South')
cond_category_elec = (df_example['Product_Category'] == 'Electronics')

# ✅ Combine with |
sum_sales_south_or_elec = df_example.loc[cond_region_south | cond_category_elec, 'Sales_Amount'].sum()

print("Total Sales for 'South' OR 'Electronics':")
print(sum_sales_south_or_elec)

Output:

Total Sales for 'South' OR 'Electronics':
5100

Using DataFrame.query() with or​

import pandas as pd

df_example = pd.DataFrame({
    'Region': ['North', 'South', 'North', 'West', 'South'],
    'Product_Category': ['Electronics', 'Books', 'Apparel', 'Electronics', 'Books'],
    'Sales_Amount': [1500, 800, 1200, 2200, 600]
})

sum_sales_south_or_elec_query = df_example.query(
    "Region == 'South' or Product_Category == 'Electronics'"
)['Sales_Amount'].sum()

print("Total Sales for 'South' OR 'Electronics' (query()):")
print(sum_sales_south_or_elec_query)

Output:

Total Sales for 'South' OR 'Electronics' (query()):
5100

Conclusion​

Summing values in a Pandas DataFrame column based on one or more conditions is a powerful way to perform targeted aggregations.

  • Boolean Indexing with df.loc[condition, 'column_to_sum'].sum() is the most idiomatic and flexible Pandas approach.
    • Use & for AND logic between multiple conditions.
    • Use | for OR logic between multiple conditions.
    • Always wrap individual conditions in parentheses: (condition1) & (condition2).
  • df.query("your_condition_string")['column_to_sum'].sum() provides a readable string-based alternative for expressing conditions.

These methods allow you to precisely define the subset of data you want to sum, enabling more insightful data analysis.

Table of Contents