主页 / 问题 / 1439174
Accepted
Alex Kazantsev
Asked:2022-08-14 16:21:35 +0000 UTC2022-08-14 16:21:35 +0000 UTC

如何有条件地在 Pandas 框架列中堆叠一些单元格？

  • 772

我有一个 Pandas 数据框，我需要在其中沿某个列添加一些行。此外，这不应随意进行，而是按顺序向下移动，测量order变量指定的行数。

所以，有多少行要计入总和，用户定义的指标顺序会告诉. 在我们的例子order=3中，这意味着我们添加了 3 行。到目前为止，我已经收到了原始表格。想总结一下。我会很感激的想法。到目前为止，我认为按行分组的方向，但我发现很难实现。

编码：

import pandas as pd

order = 3
y_0 = 0
koefs_list = [3,4,5]
parameters_list = [1,1,1,2,2,2,3,3,3]

df = pd.DataFrame({"params":parameters_list, "koefs":koefs_list*int(len(parameters_list)/len(koefs_list))})
df["res"] = df["params"]*df["koefs"]+y_0
print(df)

在输出中，我需要按列添加值res：

    params  koefs  res
0       1      3    3
1       1      4    4
2       1      5    5
3       2      3    6
4       2      4    8
5       2      5   10
6       3      3    9
7       3      4   12
8       3      5   15
python pandas

1 个回答

  1. Best Answer

    通常，您已经将数据划分为必要的三元组，您可以这样做：

    print(df.groupby('params')['res'].sum())

    但是，如果仍然无法关注这一点，那么您可以简单地计算除以整数除以的元素order：

    df.groupby(pd.Series(range(df.shape[0]))//order)['res'].sum()

    在这两种情况下，输出将是：

    0    12
1    24
2    36
Name: res, dtype: int64
    • 1

相关问题