from sklearn import datasets

# load iris dataset

iris = datasets.load_iris()

# Since this is a bunch, create a dataframe

iris_df=pd.DataFrame(iris.data)

iris_df['class']=iris.target

iris_df.columns=['sepal_len', 'sepal_wid', 'petal_len', 'petal_wid', 'class']

iris_df.dropna(how="all", inplace=True) # remove any empty lines

## Original code provided by Srishti Saha in Github