from sklearn import datasets
import pandas as pd
raw_data = datasets.load_iris()
# Load iris dataset
iris_df = pd.DataFrame(raw_data.data)
iris_df['class'] = raw_data.target.astype('str')
iris_df.columns=['sepal_length', 'sepal_width', 'petal_length', 'petal_width', 'class']
iris_df
from scipy.stats import chisquare
chisquare(f_obs = [561,551])