如何在Python中过滤用户输入的月份以用于数据分析？

嗨，我来帮你把用户输入的月份用到DataFrame筛选里！你已经完成了月份输入的验证逻辑，接下来只需要把输入的月份转换成对应的筛选条件就行，这里给你两种实用的方案：

方案1：动态生成日期范围（兼容字符串/日期类型的列） #

这种方法和你之前手动写日期范围的逻辑一致，只是把固定的月份换成用户输入的动态值，还优化了输入验证的交互体验：

import pandas as pd

# 用循环确保用户输入有效的月份，不用重新运行程序
while True:
    try:
        MonthAnalysis = int(input("Enter which Month to analyse:"))
        if 1 <= MonthAnalysis <= 12:
            break
        else:
            print("Invalid, please choose a number between 1 and 12")
    except ValueError:
        print("Please enter an integer number")

# 假设我们分析2018年的目标月份，动态生成起止日期
# 用`:02d`保证月份是两位数（比如4自动变成'04'）
start_date = f'2018-{MonthAnalysis:02d}-01'
# 单独处理12月的特殊情况，结束日期是次年1月1日
if MonthAnalysis == 12:
    end_date = '2019-01-01'
else:
    end_date = f'2018-{MonthAnalysis + 1:02d}-01'

# 执行DataFrame筛选
df = df[(df['a_date'] > start_date) & (df['a_date'] < end_date)]

方案2：用Pandas的日期属性筛选（更直观，需日期列是datetime类型） #

如果你的a_date列还不是datetime类型，先转成该类型，然后直接通过日期属性筛选，还能精准限定年份避免跨年份匹配：

import pandas as pd

# 先把日期列转换为datetime类型（如果还没转的话）
df['a_date'] = pd.to_datetime(df['a_date'])

# 同样处理用户输入的有效性
while True:
    try:
        MonthAnalysis = int(input("Enter which Month to analyse:"))
        if 1 <= MonthAnalysis <= 12:
            break
        else:
            print("Invalid, please choose a number between 1 and 12")
    except ValueError:
        print("Please enter an integer number")

# 筛选2018年指定月份的数据
df = df[(df['a_date'].dt.year == 2018) & (df['a_date'].dt.month == MonthAnalysis)]

两种方案都能实现你的需求：方案2更简洁直观，推荐优先使用；如果你的日期列是字符串格式且不想转换，方案1会更合适。

内容的提问来源于stack exchange，提问作者Tyson