Pandas 遍历行并求和

桃扇骨 2022-12-10 14:42 212阅读 0赞

# Pandas 遍历行并求和 #

*用 Pandas 生成一列是另一列的逐行累加结果*

import numpy
    import pandas
    from pandas import DataFrame

cash = pandas.read_csv('./data.csv')
    cash

<table> 
 <thead> 
  <tr> 
   <th></th> 
   <th>银行</th> 
   <th>编号</th> 
   <th>收入/支出</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <th>0</th> 
   <td>中国工商银行</td> 
   <td>001/2020</td> 
   <td>10000</td> 
  </tr> 
  <tr> 
   <th>1</th> 
   <td>中国工商银行</td> 
   <td>002/2020</td> 
   <td>50000</td> 
  </tr> 
  <tr> 
   <th>2</th> 
   <td>中国工商银行</td> 
   <td>003/2020</td> 
   <td>-30000</td> 
  </tr> 
  <tr> 
   <th>3</th> 
   <td>中国工商银行</td> 
   <td>004/2020</td> 
   <td>90000</td> 
  </tr> 
  <tr> 
   <th>4</th> 
   <td>中国工商银行</td> 
   <td>005/2020</td> 
   <td>10000</td> 
  </tr> 
 </tbody> 
</table>

cash.loc[:, '动态结余'] = cash.loc[:, '收入/支出'].cumsum()
    cash

<table> 
 <thead> 
  <tr> 
   <th></th> 
   <th>银行</th> 
   <th>编号</th> 
   <th>收入/支出</th> 
   <th>动态结余</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <th>0</th> 
   <td>中国工商银行</td> 
   <td>001/2020</td> 
   <td>10000</td> 
   <td>10000</td> 
  </tr> 
  <tr> 
   <th>1</th> 
   <td>中国工商银行</td> 
   <td>002/2020</td> 
   <td>50000</td> 
   <td>60000</td> 
  </tr> 
  <tr> 
   <th>2</th> 
   <td>中国工商银行</td> 
   <td>003/2020</td> 
   <td>-30000</td> 
   <td>30000</td> 
  </tr> 
  <tr> 
   <th>3</th> 
   <td>中国工商银行</td> 
   <td>004/2020</td> 
   <td>90000</td> 
   <td>120000</td> 
  </tr> 
  <tr> 
   <th>4</th> 
   <td>中国工商银行</td> 
   <td>005/2020</td> 
   <td>10000</td> 
   <td>130000</td> 
  </tr> 
 </tbody> 
</table>