如何在Python中编写一个函数来实现这个功能？

当然，你可以通过以下Python函数来实现这一功能。这个函数会接收两个参数：年份（year）和月份（month），然后返回指定月份的天数：

def days_in_month(year, month):
    # 定义一个包含每个月天数的字典
    month_days = {
        1: 31,
        2: 29 if (year % 4 == 0 and (year % 100 != 0 or year % 400 == 0)) else 28,
        3: 31,
        4: 30,
        5: 31,
        6: 30,
        7: 31,
        8: 31,
        9: 30,
        10: 31,
        11: 30,
        12: 31
    }

    # 检查月份的有效性（1到12）
    if month < 1 or month > 12:
        return "Invalid month"

    # 返回对应的天数
    return month_days[month]

# 测试函数
print(days_in_month(2020, 2))  # 输出：29
print(days_in_month(2021, 1))  # 输出：31

这个函数首先检查月份是否有效，然后根据给定的年份判断闰年，最后从字典中获取并返回相应月份的天数。