27 lines
695 B
Python
27 lines
695 B
Python
|
import sys
|
||
|
|
import seaborn as sns
|
||
|
|
import pandas as pd
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
|
||
|
|
def scatter(file, get_head=False):
|
||
|
|
sns.set(style="ticks", color_codes=True)
|
||
|
|
|
||
|
|
house_column = 'Hogwarts House'
|
||
|
|
first = 'Astronomy'
|
||
|
|
second = 'Defense Against the Dark Arts'
|
||
|
|
|
||
|
|
points = pd.read_csv(file)
|
||
|
|
if (get_head is True):
|
||
|
|
points = points.head(50)
|
||
|
|
only_int = points.select_dtypes(exclude=['object'])
|
||
|
|
sns.scatterplot(data=points, hue=house_column, x=first, y=second)
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
if (len(sys.argv) > 1 ):
|
||
|
|
try:
|
||
|
|
scatter(sys.argv[1], len(sys.argv) > 2 and sys.argv[2] == "-h")
|
||
|
|
except:
|
||
|
|
print("error")
|