python functions for multiple conditions

Question

below shown the sales data

i have created function so that i get the summary of sales city wise,

import pandas as pd
Super=pd.read_excel(r"C:\Script Testing\supermarket_sales.xlsx")

Location=Super[Super["City"]=="Yangon"]
def Summary():
    PDT=Location.groupby(["Product line","Invoice ID"])["Total"].sum()
    PDTT=PDT.reset_index()
    return PDT

can anyone suggest how to run this function across all the cities in the data without creating a separate function for each city

gacero · Accepted Answer · 2021-05-27 09:09:00Z

1

Following your own approach, you can add city as a variable to the function and call the function inside a loop:

def Summary(city):
    Location=Super[Super["City"] == city]
    PDT=Location.groupby(["Product line","Invoice ID"])["Total"].sum()
    PDTT=PDT.reset_index()
    return PDT

for city in Super['City'].unique():
    PDT = Summary(city)
    #whatever you do with PDT

answered May 27, 2021 at 9:09

gacero

1281 silver badge9 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

python functions for multiple conditions

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related