Assignment

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

data = pd.read_csv("testdata.csv")

data.head(10)

data.info()

data.describe()

data["site"] = data["site"].astype('category')

color = []

for lab,row in data.iterrows(): 
    if row["site"] == 2:
        color.append("red")
    elif row["site"] == 3:
        color.append("green")
    elif row["site"] == 4:
        color.append("yellow")
    else:
        color.append("black")


data["color"]=color

data.head()

plt.scatter(data["revenue"],data["top"],color = data["color"])
plt.show()