根据其他列替换值

Question

让我写一个数据框来解释我的问题：

Loc   Length   Size
A       5       50
A       10      90
A       8       0
B       20      140
B       5       50
B       12      0

假设我有一个这样的数据框。我想要做的是替换等于 0 的大小值。我想替换这个值，例如： 如果 Loc A 中的大小，我想要做的是（A 中的大小总和）/（A 中的长度总和）乘以 Length 并将该值更改为零。我想对每个等于 0 的 Size 值执行此操作，这取决于他的 Location 的 sumSize/sumLength 和他的行的 Length 值。 我试过了，但我什么也做不了。请帮我解决这个问题，谢谢！

Answer 1

IIUC groupby Loc 得到总和，然后 map 得到结果：

s = df.loc[df["Size"].ne(0)].groupby("Loc").sum()

df.loc[df["Size"].eq(0), "Size"] = df["Loc"].map(s["Size"]/s["Length"])*df["Length"]

print (df)

  Loc  Length        Size
0   A       5   50.000000
1   A      10   90.000000
2   A       8   74.666667
3   B      20  140.000000
4   B       5   50.000000
5   B      12   91.200000