Pandas读取CSV文件并提取收盘价列

使用Pandas库读取CSV文件并提取收盘价列

stock_sales = pd.read_csv(
    comp_dir / 'train.csv',
    usecols=['Data_ID', 'Time', 'Close_Price'],
    dtype={
        'Data_ID': 'int',
        #'Time': 'datetime64',
        'Close_Price': 'float32',
    },
    parse_dates=['Time'],
    infer_datetime_format=True,
)

# 取出'Close_Price'列
x = stock_sales['Close_Price']

该代码使用pd.read_csv()函数读取名为'train.csv'的CSV文件，并指定读取'Data_ID', 'Time', 'Close_Price'三列。同时，使用dtype参数指定数据类型，并将'Time'列解析为日期时间格式。最后，将'Close_Price'列赋值给变量x。