Как объединить данные одного и того же объекта в датафрейме?

Question

Мы знаем, что существует вероятность того, что гусеница станет бабочкой, а головастик - лягушкой. Данные о весе являются исключительными для одного и того же имени с другим типом (хотя бы один из них равен нулю в том же столбце) Как я могу объединить данные для того же имени, что и новый фрейм данных (сохранить только новый тип, если существует), как показано ниже.

old_df = {'name':  ['a', 'b', 'b', 'c', 'd', 'd', 'e'],
        'type': ['tiger', 'caterpillar', 'butterfly', 'dolphin', 'tadpole', 'frog', 'dog'],
        'month1_weight': ['33', '0', '0', '59', '3', '0', '0'],
        'month2_weight': ['0', '0', '20', '65', '0', '10', '2'],
        }
old_dataframe = pd.DataFrame(old_df)

old dataframe:
name      type         month1_weight  month2_weight
a         tiger        33             0
b         caterpillar  0              0
b         butterfly    0              20
c         dolphin      59             65
d         tadpole      3              0
d         frog         0              10
e         dog          0              2

new dataframe:
name      type         month1_weight  month2_weight
a         tiger        33             0
b         butterfly    0              20
c         dolphin      59             65
d         frog         3              10
e         dog          0              2

Answer 1

Использование:

new_dataframe = (
    old_dataframe.groupby("name")
    .agg({
        "type": "last",
        "month1_weight": lambda s: s.astype(int).sum().astype(str),
        "month2_weight": lambda s: s.astype(int).sum().astype(str)})
    .reset_index()
)

print(new_dataframe)

Печать:

  name       type month1_weight month2_weight
0    a      tiger            33             0
1    b  butterfly             0            20
2    c    dolphin            59            65
3    d       frog             3            10
4    e        dog             0             2