Измените имена столбцов в python pandas, используя понимание списка

Question

В настоящее время я пытаюсь изменить следующие имена столбцов:

Unnamed: 1 | Unnamed: 2         <----- Column names
    Alfa       Beta             <----- Data 

Into:

 2016 | 2017
 Alfa   Beta

Я использовал:

df.rename(columns=lambda x: re.sub(r"Unnamed\:\s\d+"," ",x))

Для замены имена столбцов с пробелом, но я хочу создать al oop, чтобы автоматически можно было заменить их на 2016 и 2017 годы

[Дополнительно] Я сделал это, чтобы создать список значений

for i in range(len(df.columns[:])):
    i+=2016
    str(i)

Результат дает:

2016
2017

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

Answer 1

df.rename(columns=lambda x: int(x.strip('Unnamed: '))+2015)

Answer 2

Я думаю, что понимание списка здесь не нужно, только присваивайте range с началом из переменной:

start = 2016
df.columns = range(start, start + len(df.columns))
print (df)
   2016  2017
0  Alfa  Beta