Python pandas循环值以两列为条件

Question

在我的数据框“数据”中，我有两列'trend'和'rtrend'

trend值为-1,0和1。

def newfunc(a):

j = -1

for i in a:

    j = j+1
    x = (j-1)

    if data.iloc[j]['trend'] != 0:

        return data.iloc[j]['trend'] 

    if data.iloc[j]['trend'] == 0:

        return data.iloc[x]['rtrend']

如果trend等于-1或1，那么我想将rtrend列值设置为trend 。

如果trend等于0，则将rtrend设置rtrend等于数据rtrend上方显示的该系列中的最后一个值。

data['rtrend'] = newfunc(data['trend'])

对于整个系列，它当前返回的全部为0。

请有人指出我正确的方向吗？ 我敢肯定必须有更好的方法来做到这一点。 （我已经尝试过np.where() ，它似乎没有做我想要的事情）。

Answer 1

不要做一个程序缓慢for循环。 做矢量化方法。 只需将非零数据复制到新的rtrend列中，然后转发填充数据：

df['rtrend'] = df[df.trend!=0]['trend']

df
Out[21]: 
   trend    b    c  rtrend
a   -1.0  1.0 -1.0    -1.0
c    0.0 -1.0  1.0     NaN
e    1.0 -1.0 -1.0     1.0
f   -1.0  1.0 -1.0    -1.0
h   -1.0  1.0  1.0    -1.0

df['rtrend'].ffill()
Out[22]: 
a   -1.0
c   -1.0
e    1.0
f   -1.0
h   -1.0
Name: rtrend, dtype: float64