根据条件减去列中的值

Question

这是我的数据框的设置：

import pandas as pd

a=pd.DataFrame([[101,'RK',1,'01','A',200],[101,'FET01',1,'02','B',10],
                [101,'CS',1,'01','C',300],[101,'AS',1,'03','D',250],
                [101,'FET02',1,'04','E',15],[102,'DG',1,'05','F',200],
                [103,'GH',3,'06','G',150],[103,'FET01',3,'08','H',12]],
                columns=['Inv','Item','Order','Warehouse','Module','Amt'])

输出：

     Inv    Item    Order   Warehouse   Module    Amt
0   101      RK       1       01          A       200
1   101    FET01      1       02          B       10
2   101      CS       1       01          C       300
3   101      AS       1       03          D       250
4   101    FET02      1       04          E       15
5   102      DG       1       05          F       200
6   103      GH       3       06          G       150
7   103    FET01      3       08          H       12

我想的。减去 Amt的的Item ，从它的相应的FET Amt 。 对于等式，由于第一行中的Item RK在其下具有Item FET01 ，因此应将RK的Amt更改为（200-10）。 在第3行中， Item CS下方没有FET ， Amt不应更改。 在第四行中，在Item AS下方有FET ，应从其FET中减去Amt （250-15）。

最终输出应为：

    Inv    Item    Order   Warehouse   Module    Amt
0   101      RK       1       01          A       190
1   101    FET01      1       02          B       10
2   101      CS       1       01          C       300
3   101      AS       1       03          D       235
4   101    FET02      1       04          E       15
5   102      DG       1       05          F       200
6   103      GH       3       06          G       138
7   103    FET01      3       08          H       12

Answer 1

你可以做

In [1333]: check = df.Item.str.startswith('FET').shift(-1).fillna(False)

In [1334]: df.loc[check, 'Amt'] = df.Amt.diff(-1)

In [1325]: df
Out[1325]:
   Inv   Item  Order  Warehouse Module    Amt
0  101     RK      1          1      A  190.0
1  101  FET01      1          2      B   10.0
2  101     CS      1          1      C  300.0
3  101     AS      1          3      D  235.0
4  101  FET02      1          4      E   15.0
5  102     DG      1          5      F  200.0
6  103     GH      3          6      G  138.0
7  103  FET01      3          8      H   12.0

---

Details

In [1336]: check
Out[1336]:
0     True
1    False
2    False
3     True
4    False
5    False
6     True
7    False
Name: Item, dtype: bool

Answer 2

您可以通过简单地遍历数据帧，检查下一行中“ Item”的值，然后减去其中是否包含“ FET”来实现此目的：

for ii, row in a.iterrows():
    if ii+1 >= len(a):
        break
    if 'FET' in a.iloc[ii+1]['Item']:
        a.set_value( ii, 'Amt', a.iloc[ii]['Amt']-a.iloc[ii+1]['Amt'] )

print a