具有其他列累积结果的系列字典

Question

我在下面的数据框中尝试创建新的列C，这样它就基于列“A”和列“B”的相对值作为字典。而且，如果“B”列中的“0”，则该键的条目将从“C”中删除

df = DataFrame({'A' : [1,2,3,2,3,2],
            'B':['Hi','Hello','HiWorld','HelloWorld','0','0']})

for indx,row in df.iterrows():
    df['C'].append(dict(zip([row['A'],row['B']])))

我在C列中查找以下输出：

   A              B             C
0  1             Hi            {1:Hi}
1  2          Hello            {1:Hi,2:Hello}
2  3        HiWorld            {1:Hi,2:Hello,3:HiWorld}
3  2     HelloWorld            {1:Hi,2:HelloWorld,3:HiWorld}
4  3              0            {1:Hi,2:HelloWorld}
5  2              0            {1:Hi}

我尝试过使用cumsum、concat和；序列。移位（1），但有块。现在我遇到了使用dict&；拉链似乎干净的解决方案，但不适合我。任何建议

Answer 1

试试这个：

d = dict()
column = list()
for _, a, b in df.itertuples():
    if b != '0':
        d[a] = b
    else:
        d.pop(a, None)
    column.append(d.copy())

df['C'] = column