代码之家 › 专栏 › 技术社区 › jovicbg

遍历dataframe中的列表并汇总其他列

dataframe loops list pandas python

jovicbg · 技术社区 · 7 年前

我有一件棘手的事要做。

Side_a  childs                             column1
1001     NaN                                  3
1002     ['1001']                             5
1003     ['1001']                             3
1004     ['1001,'1003']                       6 
1005     ['1001', '1002', '1003', '1004']     13
1006     ['1003', '1004']                     8

第2栏应该是:

side_a          childs                            column1  column2
1001       NaN                                      3         3
1002       ['1001']                                 5         8
1003       ['1001']                                 3         6
1004       ['1001','1003']                          6         12
1005       ['1001', '1002', '1003', '1004']         13        30
1006       ['1003', '1004']                         8         17

说实话,我不知道从哪里开始,怎么开始,所以任何暗示都是很受欢迎的。

2 回复 | 直到 7 年前

Franco Piccolo 7 年前

您可以使用 loc . 最后你用 apply 使用自定义函数聚合列,如:

def row_agg(row):
    column1 = row['column1']
    childs = df.loc[row['childs']]['column1'].sum() if isinstance(row['childs'],list) else 0
    return column1 + childs

df = df.set_index('Side_a')
df['column2'] = df.apply(row_agg, axis=1)

df

    childs  column1 column2
Side_a          
A   NaN         3   3
B   [A]         5   8
C   [A]         3   6
D   [A, C]      6   12
E   [A, B, C, D]13  30
F   [C, D]      8   17

Lukas Thaler 7 年前

df 在上面的输入数据帧中,下面的代码生成所需的输出

df.index = [str(x) for x in df['side_a']]
df['column2'] = df['column1'] + [sum(df['column1'].loc[df['childs'][i]]) if isinstance(df['childs'][i], list) else 0 for i in range(df.shape[0])]

推荐文章

BlurKid · R中for循环时结果的奇怪差异

11 月前

Costinean Sebastian · 如何优化使用for循环处理大型数组的MATLAB脚本的执行时间?"

11 月前

bigjdawg43 · 迭代多个数据帧中的列并有条件地执行操作

1 年前

Colin · 缩短嵌套文件夹名称

1 年前

xhamsterIT · 循环VBA Microsoft Excel

1 年前

Nico44044 · 使用for循环遍历Django模型字段

1 年前

chanbo chung · 如何在聚合中获得所有可能的组合

1 年前

Himanshu · 无法在逐行二进制搜索中迭代2D数组中的所有行

1 年前

Sonia Acosta · 如何在python for while循环中获取下一个用户输入?[副本]

1 年前

stephr · 循环为多个变量选择最接近另一个日期的日期

1 年前