как применить преобразование к каждой строке, которая использует другие строки

Question

Я пытаюсь очистить фрейм данных ниже:

col1, clo2
A,    1
A,    2
B,    3
B,    4

, и я хочу, чтобы он выглядел так:

col1, clo2
A,    1
A,    1
B,    3
B,    3

В основном сгруппируйте по col1 и поверните все значенияв col2 до первого значения каждой группы, но я хочу сохранить все строки (у них есть данные, которые я не могу потерять).

В R с использованием dplyr я бы сделал следующее вЧтобы получить тот же результат:

data %>% group_by(col1) %>% mutate(col2 = first(col2))

Answer 1

У нас есть first и transform (mutate в R) в pandas

df.clo2=df.groupby('col1').clo2.transform('first')
df
Out[1024]: 
  col1  clo2
0    A     1
1    A     1
2    B     3
3    B     3