根据条件进行两列计算

Question

根据如下规则，增加一列输出结果：
1.当mp6>=0时，输出结果为2mp3-mp6;
2.当mp6<0时，输出结果为2mp3+mp6;

ts_code trade_date mp6 mp3
300833.SZ 20211119 2.908333333 0.4
300833.SZ 20211118 5.325 1.55
300833.SZ 20211117 7.306666667 3.506666667
300833.SZ 20211116 10.37166667 5.416666667
300833.SZ 20211115 13.10833333 9.1
300833.SZ 20211112 15.15833333 11.10666667
300833.SZ 20211111 17.09833333 15.32666667
300833.SZ 20211110 17.54833333 17.11666667
300833.SZ 20211109 17.11833333 19.21
300833.SZ 20211108 15.00666667 18.87
300833.SZ 20211105 11.67333333 17.98
300833.SZ 20211104 8.796666667 15.02666667
300833.SZ 20211103 6.536666667 11.14333333
300833.SZ 20211102 4.603333333 5.366666667
300833.SZ 20211101 3.298333333 2.566666667
300833.SZ 20211029 2.926666667 1.93
300833.SZ 20211028 4.768333333 3.84
300833.SZ 20211027 5.46 4.03
300833.SZ 20211026 5.253333333 3.923333333
300833.SZ 20211025 5.666666667 5.696666667
300833.SZ 20211022 5.97 6.89
300833.SZ 20211021 5.361666667 6.583333333
300833.SZ 20211020 3.95 5.636666667
300833.SZ 20211019 2.653333333 5.05
300833.SZ 20211018 2.16 4.14
300833.SZ 20211015 0.721666667 2.263333333
300833.SZ 20211014 -1.61 0.256666667
300833.SZ 20211013 -3.643333333 0.18
300833.SZ 20211012 -4.798333333 -0.82
300833.SZ 20211011 -6.061666667 -3.476666667
300833.SZ 20211008 -6.348333333 -7.466666667
300833.SZ 20210930 -5.728333333 -8.776666667
300833.SZ 20210929 -4.011666667 -8.646666667
300833.SZ 20210928 -1.705 -5.23
300833.SZ 20210927 0.625 -2.68
300833.SZ 20210924 3.375 0.623333333
300833.SZ 20210923 4.746666667 1.82
300833.SZ 20210922 4.468333333 3.93
300833.SZ 20210917 3.491666667 6.126666667
300833.SZ 20210916 2.876666667 7.673333333

Answer 1

可以这样处理：

m=df.loc[:, ['mp6', 'mp3']]
r=[]
for i in range(len(m)):
    if m.loc[i,'mp6']>=0:
        r.append(2*m.loc[i,'mp3']-m.loc[i,'mp6'])
    else:
        r.append(2*m.loc[i, 'mp3']+m.loc[i, 'mp6'])
df['res']=r
print(df.head())

F:\2021\qa\ot2>t8
     ts_code  trade_date        mp6       mp3       res
0  300833.SZ    20211119   2.908333  0.400000 -2.108333
1  300833.SZ    20211118   5.325000  1.550000 -2.225000
2  300833.SZ    20211117   7.306667  3.506667 -0.293333
3  300833.SZ    20211116  10.371667  5.416667  0.461667
4  300833.SZ    20211115  13.108333  9.100000  5.091667