Правая строка такая же, левая строка копирует и повторяет значения

Question

У меня есть фрейм данных

index company 
ir     iris company
ir     ir company
in     india company
in     ind company
cy     cylindar company
cy     cyli company
ds     dsl company
ds     dsli company

У меня есть вышеуказанный фрейм данных. Если ir такое же, у меня есть компания, которая скопирует первый экземпляр по каждому индексу. Например, все они будут иметь компанию iris, все будет индийскую компанию, и все они будут иметь компанию цилиндров. Результат должен быть:

ir     iris company
ir     iris company
in     india company
in     india company
cy     cylindar company
cy     cylindar company
ds     dsli company
ds     dsli company

Есть ли способ сделать это через pandas? Если нет, то что может решить эту проблему?

Answer 1

Да, вы можете сделать это с помощью pandas, используя groupby().transform('first'):

df['company'] = df.groupby('index')['company'].transform('first')