text_html_output_1.png
import matplotlib.pyplot as plt

import pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)

import seaborn as sns

datapath = '/kaggle/input/amp-parkinsons-disease-progression-prediction/'
df_proteins = pd.read_csv(f'{datapath}train_proteins.csv')
df_peptides = pd.read_csv(f'{datapath}train_peptides.csv')
df_cd = pd.read_csv(f'{datapath}train_clinical_data.csv')

data = df_proteins.merge(df_peptides, on=['visit_id', 'visit_month', 'patient_id', 'UniProt'], how='left')
data = data.merge(df_cd, on=['visit_id', 'visit_month', 'patient_id'], how='left')
data

color = 'RdYlGn'

data['upd23b_clinical_state_on_medication'].unique()