如何根据用户输入的星期字符串筛选DataFrame的Start Time列记录

你的代码有两个关键问题导致返回空DataFrame，我来一步步帮你修正：

问题1：输入语句的语法错误 #

原代码里的输入行写错了，多了不必要的user_day.在提示字符串里，应该改成正确的输入调用：

user_day = input('Input the name of the day: ')

问题2：星期比较的逻辑错误 #

df['Start Time'].dt.weekday返回的是0到6的整数（对应周一到周日，0=周一，6=周日），而dt.datetime.strptime(user_day, '%A')返回的是一个datetime对象——你直接把整数和datetime对象做相等比较，结果肯定全是False，自然得到空DataFrame。

修正方案1：直接比较星期名称字符串（最直观） #

Pandas的dt.day_name()方法可以直接返回datetime列的星期名称字符串（比如"Sunday"），我们直接和用户输入的字符串对比即可：

import pandas as pd
import datetime as dt

# 假设你的df已提前定义，示例：
# df = pd.DataFrame({'Start Time': pd.date_range('2024-01-01', periods=10)})

user_day = input('Input the name of the day: ')
filtered_df = df[df['Start Time'].dt.day_name() == user_day]
print(filtered_df)

修正方案2：将输入的星期转换为对应数字再比较 #

如果你习惯用数字比较，可以把用户输入的星期名称转换为对应的整数，再和dt.weekday的结果匹配：

import pandas as pd
import datetime as dt

user_day = input('Input the name of the day: ')
# 将输入的星期名称转为datetime对象，再取weekday()得到对应数字
target_weekday = dt.datetime.strptime(user_day, '%A').weekday()
filtered_df = df[df['Start Time'].dt.weekday == target_weekday]
print(filtered_df)

额外实用提示 #

• 输入的星期名称要严格匹配英文大小写（比如必须是"Sunday"而非"sunday"），如果想忽略大小写，可以把两边都转成小写：

  filtered_df = df[df['Start Time'].dt.day_name().str.lower() == user_day.lower()]
  

• 确认你的'Start Time'列确实是datetime类型，可用df['Start Time'].dtype检查；如果不是，先转换：df['Start Time'] = pd.to_datetime(df['Start Time'])

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