Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- import pandas as pd
- dataFrame = pd.read_csv('student.csv', delimiter=';')
- x = dataFrame.corrwith(dataFrame['G3'])
- male = dataFrame.loc[dataFrame["sex"] == "M"]
- female = dataFrame.loc[dataFrame["sex"] == "F"]
- rom = dataFrame.loc[dataFrame["romantic"] == "yes"]
- nonrom = dataFrame.loc[dataFrame["romantic"] == "no"]
- print("Correlation")
- print(x.sort_values(kind="quicksort"))
- print("Male means")
- print(male.mean())
- print("Female means")
- print(female.mean())
- print("Romantic mean")
- print(rom.mean())
- print("Non romantic mean")
- print(nonrom.mean())
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement