In [11]:
#reading the file content
f = open("nfl-suspensions-data.csv")
nfl_suspensions = list(csv.reader(f))
In [13]:
print(nfl_suspensions[0])
In [14]:
nfl_suspensions.pop(0)
Out[14]:
In [15]:
print(nfl_suspensions[0])
In [16]:
years = {}
for val in nfl_suspensions:
y = val[-2]
if y in years:
year_count = years[y]
years[y] = year_count+1
else:
years[y] = 1
print(years)
In [20]:
# unique teams
unique_teams = []
for val in nfl_suspensions:
unique_teams.append(val[1])
unique_teams = set(unique_teams)
In [21]:
#unique games
unique_games = []
for val in nfl_suspensions:
unique_games.append(val[2])
unique_games = set(unique_games)
In [22]:
print(unique_teams)
print(unique_games)
In [49]:
# creating a class suspension to store the suspension details
class Suspension(object):
def __init__(self,data):
self.name = data[0]
self.team = data[1]
self.games = data[2]
try:
year = int(data[-2])
self.year = year
except Exception:
self.year = 0
def get_year(self):
return self.year
def __str__(self):
return self.name +" " +str(self.year)
In [50]:
third_suspension = Suspension(nfl_suspensions[2])
In [51]:
print(third_suspension)
In [52]:
missing_year = Suspension(nfl_suspensions[22])
In [53]:
print(missing_year)
In [55]:
print(missing_year.get_year())
No comments :
Post a Comment