python rearrange columns pandas code example

Example 1: pandas reorder columns

# setting up a dummy dataframe
raw_data = {'name': ['Willard Morris', 'Al Jennings', 'Omar Mullins', 'Spencer McDaniel'],
        'age': [20, 19, 22, 21],
        'favorite_color': ['blue', 'red', 'yellow', "green"],
        'grade': [88, 92, 95, 70]}
df = pd.DataFrame(raw_data, index = ['Willard Morris', 'Al Jennings', 'Omar Mullins', 'Spencer McDaniel'])
df

#now 'age' will appear at the end of our df
df = df[['favorite_color','grade','name','age']]
df.head()

Example 2: rearrange columns pandas

You could also do something like this:

df = df[['mean', '0', '1', '2', '3']]
You can get the list of columns with:

cols = list(df.columns.values)
The output will produce:

['0', '1', '2', '3', 'mean']