用datetime库输出指定星期的所有日期（如输出2020年5月是星期六和星期日的日期）

用datetime库输出指定星期的所有日期（如输出2020年5月是星期六和星期日的日期）

import datetime
import calendar

y = 2020
m = 3
week_list = [6,7]  # 1为周一,7为周日
cal = calendar.month(y,m)
print(cal)

ndays = (datetime.date(y, m + 1, 1) - datetime.date(y, m, 1)).days
date_list = []
for i in range(1,ndays+1):
    p = datetime.date(y,m,i)
    nweek = p.isoweekday() #返回数字1-7，1为周一,7为周日
    if nweek in week_list:    
        d2 = p.strftime('%Y%m%d')
        date_list.append(d2)
print(date_list)

思路：
首先确定这个月的天数ndays，
然后从1开始遍历每一天，判断这一天的星期数nweek，
最后将符合条件的日期添加到列表date_list中。