pandas中DataFrame利用loc截取部分数据,end_date和start_date

本文介绍了一种在Python中使用pandas库处理股票历史数据的技巧,特别关注于如何正确地切片DataFrame以获取指定日期范围内的数据。通过实例演示了当数据按日期逆序排列时,应如何调整切片操作以避免返回空数据帧。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

def attribute_daterange_history(security,start_date,end_date,fields=('open','close','high','low')):
        daily = pro.daily(ts_code=security)
        daily.to_csv(security+'.csv')
        daily_cal = pd.read_csv(security+'.csv',index_col='trade_date')
        daily_cal = daily_cal.loc[end_date:start_date,['open','close','high','low']]
        return daily_cal
print(attribute_daterange_history('000001.SZ',20181120,20181130))

其中daily_cal = daily_cal.loc[end_date:start_date,['open','close','high','low']],红字部分如果替换成start_date:end_date就会报错。

daily_cal = daily_cal.loc[end_date:start_date,['open','close','high','low']]运行结果:

             open  close   high    low
trade_date                            
20181130    10.22  10.36  10.36  10.17
20181129    10.35  10.20  10.39  10.19
20181128    10.17  10.28  10.31  10.16
20181127    10.35  10.18  10.39  10.16
20181126    10.34  10.34  10.46  10.30
20181123    10.53  10.32  10.58  10.31
20181122    10.67  10.55  10.67  10.48
20181121    10.50  10.61  10.67  10.48
20181120    10.75  10.57  10.80  10.52

daily_cal = daily_cal.loc[start_date:end_date,['open','close','high','low']],运行结果为:

Empty DataFrame
Columns: [open, close, high, low]
Index: []

这与导入数据trade_date的排序方式有关。在此例中,数据从大至小排列。因此应将end_date放置在前。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值