如何在Pandas中拆分系列并将其应用于列

Question

我正在尝试拆分列中的系列并将其映射到唯一值(在本例中为user)。

到目前为止，我已经能够使数据帧看起来像这样。

df1:
user      weight
0  james     [100.0]
1  brandon   [60.0, 40.0]
2  brandon   [60.0, 40.0]
3  chris     [100.0]
4  james     [80.0, 5.0, 15.0]      
5  james     [80.0, 5.0, 15.0]
6  james.    [80.0, 5.0, 15.0]

我尝试了拆分爆炸方法，但它创建了重复的行。

按照期望的输出循环遍历序列并更新数据帧的最佳方法是什么?

df2:
0 james    [100.0]
1 brandon  [60.0]
2 brandon  [40.0]
3 chris    [100.0]
4 james    [80.0]
5 james    [5.0]
6 james    [15.0]

谢谢。

Answer 1

您可以使用drop_duplicates和explode

>>> df.drop_duplicates('user').explode('weight')
user weight
0    james  100.0
1  brandon   60.0
1  brandon   40.0
3    chris  100.0
6   james.   80.0
6   james.    5.0
6   james.   15.0