[Usefull stuff related to columns on Pandas] #pandas

working_with_columns.py

# Find column index
df.columns.get_loc('col_name')

# The other way around (find column name using the index)
df.columns.get_values()[index]

# Find indices of multiple columns
def column_index(df, query_cols):
    ``` Get indices of multiple columns. 
        https://stackoverflow.com/a/38489403
    ``` 
    cols = df.columns.values
    sidx = np.argsort(cols)
    return sidx[np.searchsorted(cols,query_cols,sorter=sidx)]

# Alternative with list comprehension
[df.columns.get_loc(c) for c in cols if c in df]