如何在具有非常数雅可比的Pandas数据框架上集成

hgb9j2n6  于 2023-04-28  发布在  其他
关注(0)|答案(1)|浏览(132)

我有一个pandas数据框架,看起来像这样:

E     p     S
.1   .1     4
.1.  .2     5
.    .      .
.    .      .
.3   .1     2
.3.  .2     4
.    .      .
.    .      .
15.  .1     5
15.  .2     9
.    .      .
.    .      .

正如你所看到的,对于E的每个值,都有多个pS的值。我试图对每个E值进行数值积分$n(E) = \Int dp p^2 S$。不幸的是,由于积分中的p^2因子,我认为我不能使用groupby(['E']).sum(),因为这将只执行S的求和,而不需要p^2因子。任何以Python方式做这个积分的方法都是值得赞赏的。

yhived7q

yhived7q1#

使用numpy's trapezoidal formula,您可以执行以下操作:

df.groupby('E').apply(lambda x: np.trapz(x['S'].mul(x['p']**2), x['p']))

相关问题