Andrews plots in pandas of Rdatasets with changed column order

andrews_curve_column_order.py

import pandas as pd
import statsmodels.api as sm

#Change next two lines for dataset, such as in
#http://vincentarelbundock.github.io/Rdatasets/

data = sm.datasets.get_rdataset('airquality').data
class_column = 'Month'

fig, (ax1, ax2) = plt.subplots(nrows=2, ncols=1, sharex=True)

#Plot w/ original column order
andrews_curves(data, class_column=class_column, ax=ax1)

#Rearrange columns
cols = data.columns.tolist()
cols = cols[-1:] + cols[:-1]
data = data[cols]

#Plot w/ changed column order
andrews_curves(data, class_column=class_column, ax=ax2)

ax1.legend().set_visible(False)
ax2.legend().set_visible(False)
plt.show()

Note: this is not a shuffle of column order but a rotation of column order. For a shuffle, use something like this:

import random
cols = data.columns.tolist()
random.shuffle(cols)
data = data[cols] 

Hi @amanahuja. How could I get my hands on the implementation used on line 13 for n-dimensional data?

Thanks in advance ;).