Codecademy export

plot.py

import matplotlib.pyplot as plt
import pandas as pd
import seaborn as sns
import codecademylib3

# load in the data
df = pd.read_csv("mushroom_data.csv")
print(df.head())

# list of all column headers
#columns = df.columns.str.strip()
columns = df.columns.tolist()

for column in columns:
  print(column)
  sns.countplot(df["column"])
  plt.show()
  plt.clf()