python - 如何在数据框中按顺序添加值

Question

我正在研究土地消费和人口增长。我有一个数据框，其中列出了一段时间的人口时间序列。

import pandas as pd
df = pd.DataFrame({'year': [2014, 2015, 2016, 2017, 2018], 'population': [66354, 63322,83381, 91563, 93709]})

我将建造的建筑物的数量按年份分组，这是徒劳的。

df2 = pd.DataFrame({'year': [2014, 2015, 2016, 2017, 2018], 'numberOfBu': [1348, 900, 1137, 865, 235]})

为了比较建筑物和人口之间的顺序，我应该逐年对建筑物进行增量添加 - 就像人口值一样。所以我必须将第一个值与第二个值相加，将两者的结果与第三个值相加，依此类推。

有熊猫的解决方案吗？

标签： pythonpandasdataframe

Answer 1

IIUC 您正在搜索累积和函数cumsum()：

df.population.cumsum()
# 0     66354
# 1    129676                                               
# 2    213057                                           
# 3    304620                                               
# 4    398329                                                
# Name: population, dtype: int64