Combine a variable and a string for naming a column in pandas

Question:

I have a dataframe with several column which have similar names, something like col_1, col_2, col_3.

I am trying to create a for loop to go over the 3 columns and want to use .loc inside the loop to slice the dataframe.

Something like this:

vars = [1, 2, 3]

for i in vars:
    df.loc[:, ('col_'+ i):]

What is the correct way to do it?

Asked By: arkon

||
Source

Answers:

Have you considered using f-strings?

vars = [1, 2, 3]

for i in vars:
    df.loc[:, f'col_{i}']
Answered By: rpanai