大家用 pandas dataframe 有什么好的跨行( row)计算的方法?

2016-08-27 20:24:30 +08:00
 gubo2012
举个简单的例子, 1,2,3 三行, A,B,C 三列,要求C(n+1) = A(n) + B(n),n 表示行数,这个有什么好的实现方法?

有个比较笨的方法是, 1 )加一列 D ,记录行数, 2 )再加一列 E , E = D +1,3)克隆一个 df,4)将两个 df   merge ,让第一个 df 的D列等于第二个 df 的E列,然后计算计算A+B。这样可行,但太麻烦。

请问有没有更好的方法, loop 除外。谢谢
4048 次点击
所在节点    Python
2 条回复
lkx4102
2016-08-27 20:37:41 +08:00
df.['c'] = df.['b'].shift(1) + df.['a'].shift(1)
zhuangzhuang1988
2016-08-27 21:55:19 +08:00
试试矩阵乘法。。

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/302178

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX