Question

我想用python熊猫分析历史股票市场数据上的交易算法。但是我了解到，使用大型数据集循环并不是真的很快-数百万行是不可能的。

因此，我从布尔索引开始。但是我无法正常工作。有人知道吗？我的示例有5行，但实际上我有200万行。

我了解了SHIFT函数，以考虑上一行的值。但这仅能解决此问题。

## data set
timehourminute=['15:25','15:30','15:35','15:40','15:45']
close=[21.02,21.05,21.10,21.22, 21.17]
signal=[False,True,True,True,False]
position=[0,0,0,0,0]
data={'timehourminute':timehourminute,'close':close, 'signal':signal,'position':position}
df=pd.DataFrame.from_dict(data)

## if time = 15:30 and signal = True, buy $1000 worth of stocks
subset = (df.timehourminute=='15:30') & (df.signal==True)
df.loc[subset,'position']=(1000/df.close)

## if previous row has position, keep the position if the signal is still True
df['positionprev']=df.position.shift(1)
df.position = df.signal * df[['position','positionprev']].max(axis=1)
df.position = df.position.astype(int)

输出是这样的：

timehourminute  close  signal  position  positionprev
0          15:25  21.02   False         0           NaN
1          15:30  21.05    True        47      0.000000
2          15:35  21.10    True        47     47.505938
3          15:40  21.22    True         0      0.000000
4          15:45  21.17   False         0      0.000000

问题是我想保持位置直到信号变为假。在此示例中，信号持续3行，但实际上可能持续1000行。我如何获得47的值以保持在第3行中？

Answer 1

我相信您需要Series.mask来按条件获取缺失值并向前填充缺失值：

df['new'] = df.position.mask(df.signal & (df.position == 0)).ffill()
print (df)
  timehourminute  close  signal   position        new
0          15:25  21.02   False   0.000000   0.000000
1          15:30  21.05    True  47.505938  47.505938
2          15:35  21.10    True   0.000000  47.505938
3          15:40  21.22    True   0.000000  47.505938
4          15:45  21.17   False   0.000000   0.000000

关于为股市算法循环熊猫数据框的问题

1 个答案: