text_html_output_1.png
import matplotlib.pyplot as plt

import pandas as pd

df_train = pd.read_csv('/kaggle/input/playground-series-s3e13/train.csv')
df_test = pd.read_csv('/kaggle/input/playground-series-s3e13/test.csv')
df_orig_train = pd.read_csv('/kaggle/input/vector-borne-disease-prediction/trainn.csv')
df_orig_test = pd.read_csv('/kaggle/input/vector-borne-disease-prediction/testt.csv')

df_train.columns

primary_columns = list(df_train.drop(columns=['id', 'prognosis']).columns)

train = df_train.copy().drop(columns=['id'])
marginals = train.groupby('prognosis').mean()
plt.xticks(range(64), marginals.columns, rotation=90)
plt.yticks(range(11), marginals.index)
plt.colorbar()

df_train.head()