如何让datetime.fromisoformat根据输入自动输出日期或日期时间？

datetime.fromisoformat 方法返回的始终是 datetime 实例，即便输入只有日期，也会默认附加 00:00:00 时间部分。要实现你期望的输出效果，需要根据输入字符串的结构做针对性处理，以下是两种可行方案：

方案1：根据输入返回date或datetime对象 #

如果希望无时间输入时得到纯日期对象（date类型），带时间输入时保留datetime对象，可以编写一个判断处理函数：

from datetime import datetime, date

def parse_date_input(date_str):
    if 'T' not in date_str:
        # 输入为纯日期，转换为date对象
        return datetime.fromisoformat(date_str).date()
    else:
        # 输入带时间部分，保留datetime对象
        return datetime.fromisoformat(date_str)

# 测试用例
date_str = "2020-01-01"
date_obj = parse_date_input(date_str)
print(date_obj)  # 输出: 2020-01-01

date_str = "2020-01-01T20"
date_obj = parse_date_input(date_str)
print(date_obj)  # 输出: 2020-01-01 20:00:00

date_str = "2020-01-01T00:00:00"
date_obj = parse_date_input(date_str)
print(date_obj)  # 输出: 2020-01-01 00:00:00

方案2：统一返回datetime对象，格式化输出对应格式 #

如果需要保持返回对象为datetime类型，但输出字符串时匹配需求格式，可以在格式化时根据输入结构选择对应模板：

from datetime import datetime

def format_date_output(date_str):
    dt = datetime.fromisoformat(date_str)
    if 'T' not in date_str:
        # 纯日期输入，输出纯日期字符串
        return dt.strftime("%Y-%m-%d")
    else:
        # 带时间输入，输出带时间的字符串
        return dt.strftime("%Y-%m-%d %H:%M:%S")

# 测试用例
date_str = "2020-01-01"
print(format_date_output(date_str))  # 输出: 2020-01-01

date_str = "2020-01-01T20"
print(format_date_output(date_str))  # 输出: 2020-01-01 20:00:00

date_str = "2020-01-01T00:00:00"
print(format_date_output(date_str))  # 输出: 2020-01-01 00:00:00

关键说明 #

• datetime对象本身必然包含日期和时间两部分，无法直接让它"只保留日期"，所以必须通过转换为date对象，或者格式化输出时忽略时间部分来实现需求。
• 判断输入是否包含'T'是区分纯日期和带时间输入的核心依据，符合ISO 8601格式的规则。

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