如何将输入指定为日期时间？

Question

我的项目涉及读取有关城市数据的 CSV 文件，然后计算各种统计数据。我正在尝试创建一个函数，该函数将从 CSV 获取城市数据并创建数据的副本，但应用了日期过滤器。提示用户输入开始和结束日期。

我目前正在测试并尝试使用正确的日期过滤器打印数据。到目前为止，我最大的问题是我似乎无法将用户输入的字符串转换为日期时间格式。有没有办法让用户直接输入日期时间格式的数据？

def date_filtering(city_file):
    date1 = input('Please enter a valid start date that you wish to explore?\n'
    date2 = input('Please enter a valid end date.')    
    print(city_file.loc[date1:date2])

提前致谢！

Answer 1

我假设您只需要日期，因为您没有在代码中提及时间。您需要做的是以某种方式解析输入以获取年、月和日。我确信它可以使用正则表达式来完成，但只使用 python 内置函数更简单。

在美国表示日期的正常方式是 mm/dd/yyyy，所以我将使用它。

def strtodatetime(str):
     month, day, year = [int(d) for d in str.split("/")]
     return datetime.date(year, month, day)

date1 = strtodatetime(input('Please enter a valid start date that you wish to explore?\n'))
date2 = strtodatetime(input('Please enter a valid end date.'))

strtodatetime 按“/”字符分割输入字符串，然后将结果列表中的每个元素转换为整数。然后它返回一个带有相应日期的新 datetime.date 对象。

如果您担心用户输入了错误格式的内容，您可以使用 try 块来确保他们没有。我可以看到三种可能的异常来源，用户可以输入太多或太少的 /（如 10/2/1999/2），用户可以输入不存在的日期（如 13/32 /-1)，或者用户可以输入无法转换为 int 的内容（例如 March/22/1999）。所有三个都会引发 ValueErrors，所以这是我们需要在这里捕获的唯一异常。为了捕捉到这一点，我们将代码更改为

def strtodatetime(str):
     month, day, year = [int(d) for d in str.split("/")]
     return datetime.date(year, month, day)

date_filtering(city_file):
    try:
        date1 = strtodatetime(input('Please enter a valid start date that you wish to explore?\n'))
        date2 = strtodatetime(input('Please enter a valid end date.'))
    except ValueError:
        print('Please enter a valid date in the format mm/dd/yyyy')
        return date_filtering(city_file)
    return city_file.loc[date1:date2]

print(data_filtering(city_file))