问题
我的代码收到以下错误消息。可能的问题是,我先用.loc切片了数据帧,然后尝试为该切片分配值。根据我的理解,Pandas无法百分百确定我是否只想将值分配给切片,还是让其一直传播到原始df。我不确定该如何解决。
错误消息
C:\blp\BQuant\environments\bqnt-1.25.2\lib\site-packages\pandas\core\indexing.py:140: SettingWithCopyWarning:
A value is trying to be set on a copy of a slice from a DataFrame
请参阅文档中的警告:http://pandas.pydata.org/pandas-docs/stable/indexing.html#indexing-view-versus-copy
self._setitem_with_indexer(indexer, value)
完整代码
import numpy as np
import pandas as pd
import datetime as dt
import time
csv1 = pd.read_csv('stock_price.csv', delimiter = ',')
df = pd.DataFrame(csv1)
df['delta'] = df.PX_LAST.pct_change()
df.loc[df.index[0], 'avg_gain'] = 0
for x in range(1,len(df.index)):
if df["delta"].iloc[x] > 0:
df["avg_gain"].iloc[x] = ((df["avg_gain"].iloc[x - 1] * 13) + df["delta"].iloc[x]) / 14
else:
df["avg_gain"].iloc[x] = ((df["avg_gain"].iloc[x - 1] * 13) + 0) / 14
df
输入
Dates,PX_LAST
03/09/2018,157.512
04/09/2018,155.393
05/09/2018,154.069
06/09/2018,155.109
07/09/2018,156.301
10/09/2018,156.717
11/09/2018,157.19
12/09/2018,157.549
13/09/2018,159.157
14/09/2018,158.363
17/09/2018,158.968
输出
Dates,PX_LAST,delta,avg_gain
03/09/2018,157.512,NaN,0
04/09/2018,155.393,-0.013453,0
05/09/2018,154.069,-0.00852,0
06/09/2018,155.109,0.00675,0.000482
07/09/2018,156.301,0.007685,0.000997
10/09/2018,156.717,0.002662,0.001116
11/09/2018,157.19,0.003018,0.001251
12/09/2018,157.549,0.002284,0.001325
13/09/2018,159.157,0.010206,0.00196
14/09/2018,158.363,-0.004989,0.00182
17/09/2018,158.968,0.00382,0.001963
问题所在的代码行
for x in range(1,len(df.index)):
if df["delta"].iloc[x] > 0:
df["avg_gain"].iloc[x] = ((df["avg_gain"].iloc[x - 1] * 13) + df["delta"].iloc[x]) / 14
else:
df["avg_gain"].iloc[x] = ((df["avg_gain"].iloc[x - 1] * 13) + 0) / 14
解决方案
我尝试使用.copy()
,但仍然收到相同的错误消息
for x in range(1,len(df.index)):
if df["delta"].iloc[x] > 0:
df["avg_gain"].iloc[x] = ((df["avg_gain"].iloc[x - 1].copy() * 13) + df["delta"].iloc[x].copy()) / 14
else:
df["avg_gain"].iloc[x] = ((df["avg_gain"].iloc[x - 1].copy() * 13) + 0) / 14
谢谢
答案 0 :(得分:1)
问题代码可以替换为
for x in range(1,len(df.index)):
if df["delta"].iloc[x] > 0:
df.iloc[x, -1] = ((df["avg_gain"].iloc[x - 1] * 13) + df["delta"].iloc[x]) / 14
else:
df.iloc[x,-1] = ((df["avg_gain"].iloc[x - 1].copy() * 13) + 0) / 14
这是因为您最后添加了avg_gain
,因此可以使用iloc[:,-1]
访问该列。
使用ewm
更新:
arg = df["delta"].clip(lower=0)
arg.iloc[0] = 0
df['avg_gain'] = arg.ewm(alpha=1/14, adjust=False).mean()
输出:
0 0.000000
1 0.000000
2 0.000000
3 0.000482
4 0.000997
5 0.001116
6 0.001251
7 0.001325
8 0.001960
9 0.001820
10 0.001962
Name: delta, dtype: float64