如何在Python中将CSV中的日期输入保存为变量

推荐两种实用方案，根据你的需求选择：

方案1：用Python内置csv模块（无需额外安装） #

适合轻量场景，不用装第三方库：

import csv
from datetime import datetime

# 替换为你的CSV文件路径
with open('your_data.csv', 'r', encoding='utf-8') as csv_file:
    csv_reader = csv.reader(csv_file)
    # 跳过表头（如果CSV没有表头，就删掉这行）
    next(csv_reader)
    
    # 读取第一行目标列的日期（假设目标是第2列，索引从0开始所以写[1]，按需修改）
    first_entry = next(csv_reader)
    date1_raw = first_entry[1]
    # 转成datetime对象方便后续计算，格式要和CSV里的日期匹配，比如'%m/%d/%Y'
    date1 = datetime.strptime(date1_raw, '%Y-%m-%d')
    
    # 读取第二行目标列的日期
    second_entry = next(csv_reader)
    date2_raw = second_entry[1]
    date2 = datetime.strptime(date2_raw, '%Y-%m-%d')

# 现在date1和date2就是你要的变量，直接用就行
print(f"第一个日期：{date1}")
print(f"第二个日期：{date2}")

方案2：用pandas库（更简洁高效） #

如果经常处理表格数据，pandas会省很多事，先安装：pip install pandas

import pandas as pd

# 读取CSV，只加载目标列（替换成你的列名，比如'Date'）
df = pd.read_csv('your_data.csv', usecols=['Date'])

# 提取前两行的日期，自动转成datetime类型
date1 = df.iloc[0]['Date']
date2 = df.iloc[1]['Date']

# 验证变量
print(f"第一个日期变量：{date1}")
print(f"第二个日期变量：{date2}")

关键注意点 #

• 把代码里的your_data.csv换成你实际的文件路径
• 日期格式要对应：如果CSV里是05/20/2024，就把strptime的格式改成'%m/%d/%Y'
• 如果CSV没有表头，方案1删掉next(csv_reader)；方案2加个参数header=None，然后用索引取列，比如df.iloc[0][0]代表第一列第一行

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