Grouping by Several Columns

Let's add some information on the .groupby() method. You can group by several columns, but the order is crucial in this case. In the previous chapter, we grouped data by the flight number and counted the number of delays. We can make this task complicated by grouping not only by the 'Flight' column, but also by the column 'Airline'. Refresh the information on the dataset and then look at this simple example (the output contains only the first 10 rows):

import pandas as pd

data = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/4bf24830-59ba-4418-969b-aaf8117d522e/plane', index_col = 0)

data_flights = data[['Flight', 'Delay', 'Airline']].groupby(['Flight', 'Airline']).count()

print(data_flights.head(10))


              1234
            
import pandas as pd
data = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/4bf24830-59ba-4418-969b-aaf8117d522e/plane', index_col = 0)
data_flights = data[['Flight', 'Delay', 'Airline']].groupby(['Flight', 'Airline']).count()
print(data_flights.head(10))

Explanation:

python
data[['Flight', 'Delay', 'Airline']].groupby(['Flight', 'Airline']).count()

data[['Flight', 'Delay', 'Airline']] - columns you will work with, including the columns by which you will group;
.groupby(['Flight', 'Airline']) - here, 'Flight' and 'Airline' are arguments of the function .groupby().

Pay attention; if you want to group by several columns, put them into the list - the order is crucial. So, in our case, if rows of the data set have the same value in the column 'Flight', they will relate to one group. Then inside those groups, the function finds other groups for rows with the same value in the column 'Airline'. Then, due to the method .count() that counts the rows, our function will calculate the number of rows in the column 'Delay' that have the same value in the column 'Airline' for each 'Flight' group.

Opgave

Swipe to start coding

Your task here is to group data by the airport from which the flight started and then by the weekday. Calculate the average time for the groups. Follow the algorithm to manage the task:

Group data:
- Extract the columns 'AirportFrom', 'DayOfWeek', and 'Time' from data (in this order);
- Apply the .groupby() method to the previous columns;
- Within the .groupby() method, put the columns 'AirportFrom' and 'DayOfWeek'; the order is crucial;
- Calculate the mean value of the column 'Time'.
Output the first 10 rows of the data_flights.

Løsning

import pandas as pd

data = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/4bf24830-59ba-4418-969b-aaf8117d522e/plane', index_col = 0)

# Group data

data_flights = data[['AirportFrom', 'DayOfWeek', 'Time']].groupby(['AirportFrom', 'DayOfWeek']).mean()

# Output the first 10 rows of the data set

print(data_flights.head(10))

Var alt klart?

Tak for dine kommentarer!

Sektion 4. Kapitel 2

import pandas as pd

data = pd.read_csv('https://codefinity-content-media.s3.eu-west-1.amazonaws.com/4bf24830-59ba-4418-969b-aaf8117d522e/plane', index_col = 0)

# Group data

data_flights = ___[['___', '___', ___]].___(['AirportFrom', ___]).___

# Output the first 10 rows of the data set

print(data_flights.___(___))

Spørg AI

Spørg om hvad som helst eller prøv et af de foreslåede spørgsmål for at starte vores chat