Как добавлять значения с определённой строки?

Имеется следующая проблема: есть выгрузка порядка 800строк, знаю, что до 550 строки идёт 2019 год, а после 550 строки 2020. В датафрейме изначально не было столбца с годами, требуется добавить года. Собственно вопрос, как добавлять значения до определённой строки? Задача вроде простая, но не знаю как реализовать задуманное. Думаю сделать так, но везде выводит 2020

def col_year(year):
    if len(str(df)) <550:
        normal_year= str(year) + '19'
    else:
        normal_year = str(year) + '20’
    return str(normal_year)

Может быть найти один определенный индекс и от него изменять все? Прошу помощи!


Ответы (1 шт):

Автор решения: Андрей Коробейников

Поидее вам нужно просто список с годами, первые 550 элементов списка со значением 2019, остальные 250 - 2020

col_year = ['2019']*550 + ['2020']*250

И затем добавить его в DataFrame

df['col_year'] = col_year 
→ Ссылка