from datetime import date, timedelta

def days_in_month(year, month):
    return filter(lambda dt: dt.month == month, [date(year, month, 1) + timedelta(days=i) for i in range(31)])