text: import pandas as pd
  import seaborn as sns
  import matplotlib.pyplot as plt
  import numpy as np
  
  # Import data
  df = df = pd.read_csv('medical_examination.csv')
  
  # Add 'overweight' column
  df['overweight'] = df['overweight'] = df["weight"] / ((df["height"] / 100) ** 2)
  df['overweight'] = df['overweight'] = pd.Series(np.where(df['overweight'] > 25, 1, 0))
  
  # Normalize data by making 0 always good and 1 always bad. If the value of 'cholestorol' or 'gluc' is 1, make the value 0. If the value is more than 1, make the value 1.
  df['cholesterol'] = pd.Series(np.where(df['cholesterol'] > 1, 1, 0))
  df['gluc'] = pd.Series(np.where(df['gluc'] > 1, 1, 0))
  
  # Draw Categorical Plot
  def draw_cat_plot():
      # Create DataFrame for cat plot using `pd.melt` using just the values from 'cholesterol', 'gluc', 'smoke', 'alco', 'active', and 'overweight'.
      df_cat = pd.melt(df, id_vars=["cardio"], value_vars=[
                       "active", "alco", "cholesterol", "gluc", "smoke", "overweight"])
  
  
      # Group and reformat the data to split it by 'cardio'. Show the counts of each feature. You will have to rename one of the collumns for the catplot to work correctly.
      df_cat["total"] = 0
      df_cat = df_cat.groupby(["variable", "cardio", "value"], as_index=False).count()
  
      # Draw the catplot with 'sns.catplot()'
      fig = sns.catplot(x="variable", y="total", col="cardio", hue="value", kind="bar", data=df_cat).fig
  
      # Do not modify the next two lines
      fig.savefig('catplot.png')
      return fig
  
  
  # Draw Heat Map
  def draw_heat_map():
      # Clean the data
      df_heat = df.loc[
        (df['ap_lo'] <= df['ap_hi']) &
        (df['height'] >= df['height'].quantile(0.025)) &
        (df['height'] <= df['height'].quantile(0.975)) &
        (df['weight'] >= df['weight'].quantile(0.025)) &
        (df['weight'] <= df['weight'].quantile(0.975))
      ]
  
      # Calculate the correlation matrix
      corr = df_heat.corr()
  
      # Generate a mask for the upper triangle
      mask = np.triu(np.ones_like(corr, dtype=np.bool))
  
      # Set up the matplotlib figure
      fig, ax = plt.subplots(figsize=(15, 15))
  
      # Draw the heatmap with 'sns.heatmap()'
      ax = sns.heatmap(
        corr,
        mask=mask,
        fmt='.1f',
        annot=True,
        center=0,
        vmax=0.32,
        square=True,
        linewidths=.5,
        cbar_kws={"shrink": .5}
      )
      ax.set_xticklabels(ax.get_xticklabels(), rotation=90)
  
      # Do not modify the next two lines
      fig.savefig('heatmap.png')
      return fig
  