Skip to content
Course outline

Chapter 2: Data Wrangling

Working with Tables in pandas

Loading, selecting, filtering and aggregating tabular data with pandas.

Loading data

import pandas as pd

df = pd.read_csv("students.csv")
df.head()

Selecting and filtering

df[["name", "score"]]          # select columns
df[df["score"] >= 8.0]         # filter rows

Group and aggregate

The mean score of group gg is xˉg=1ng∑i∈gxi\bar{x}_g = \frac{1}{n_g}\sum_{i \in g} x_i:

df.groupby("class")["score"].mean()