Skip to content

Instantly share code, notes, and snippets.

@hardingnj
Created August 25, 2021 08:34
Show Gist options
  • Save hardingnj/a7b42b3e1a4d18d3fcd3a5098a906700 to your computer and use it in GitHub Desktop.
Save hardingnj/a7b42b3e1a4d18d3fcd3a5098a906700 to your computer and use it in GitHub Desktop.
# some code to work out how many daycare days are in a particular month.
# params
start_string = "01-09-2021"
days_to_count = 2, 3, 4 # NB monday is 0, sunday 6.
time_period = "365d"
import pandas as pd
start = pd.to_datetime(start_string, dayfirst=True)
v = pd.DataFrame({"Date": pd.date_range(start, start + pd.Timedelta(time_period))})
v["month"] = v.Date.map(lambda y: y.month_name())
v["year"] = v.Date.map(lambda y: y.year)
v["isWTF"] = v.Date.map(lambda y: y.day_of_week in days_to_count)
v.groupby(["month", "year"]).isWTF.sum()
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment