Эквивалентность функции shift() (Python) и функции lead() (R)

Общий вопрос о понимании функции SHIFT в Python:

Я начинаю с этого примера набора данных:

start = structure(list(id = c(111L, 111L, 111L, 111L, 222L, 222L, 222L
), year = c(2010L, 2011L, 2012L, 2013L, 2018L, 2019L, 2020L),
col2 = c("A", "BB", "A", "C", "D", "EE", "F"), col3 = c(242L,
213L, 233L, 455L, 11L, 444L, 123L), col4 = c(1213L, 5959L,
9988L, 4242L, 333L, 1232L, 98L)), class = "data.frame", row.names = c(NA,
-7L))

Вот что я хочу сделать в R:

library(dplyr)

end <- start %>%
mutate(year_end = lead(year),
     col2_end = lead(col2),
     col3_end = lead(col3),
     col4_end = lead(col4)) %>%
mutate_at(vars(ends_with("_end")), ~ifelse(is.na(.), "END", .)) %>%
rename(year_start = year,
     col2_start = col2,
     col3_start = col3,
     col4_start = col4)

Теперь я пытаюсь сделать то же самое в Python.

Я читал, что в Python есть функция SHIFT, которая похожа на функцию LEAD в R — вот моя попытка воссоздать эту работу в Python:

import pandas as pd

start = pd.DataFrame({
    'id': [111, 111, 111, 111, 222, 222, 222],
    'year': [2010, 2011, 2012, 2013, 2018, 2019, 2020],
    'col2': ['A', 'BB', 'A', 'C', 'D', 'EE', 'F'],
    'col3': [242, 213, 233, 455, 11, 444, 123],
    'col4': [1213, 5959, 9988, 4242, 333, 1232, 98]
})

end = start.assign(
    year_end=start['year'].shift(-1),
    col2_end=start['col2'].shift(-1),
    col3_end=start['col3'].shift(-1),
    col4_end=start['col4'].shift(-1)
).fillna('END')

end = end.rename(columns = {
    'year': 'year_start',
    'col2': 'col2_start',
    'col3': 'col3_start',
    'col4': 'col4_start'
})

Я думаю, что это выглядит разумно, но я надеялся получить вторую пару глаз, чтобы проверить мою попытку. Есть предположения?