43 lines
987 B
Python
43 lines
987 B
Python
"""
|
|
cse6242 s21
|
|
wrangling.py - utilities to supply data to the templates.
|
|
|
|
This file contains a pair of functions for retrieving and manipulating data
|
|
that will be supplied to the template for generating the table. """
|
|
import csv
|
|
|
|
def username():
|
|
return 'icanlapan3'
|
|
|
|
def data_wrangling():
|
|
with open('data/movies.csv', 'r', encoding='utf-8') as f:
|
|
reader = csv.reader(f)
|
|
table = list()
|
|
# Feel free to add any additional variables
|
|
...
|
|
|
|
# Read in the header
|
|
for header in reader:
|
|
break
|
|
|
|
count = 0
|
|
|
|
# Read in each row
|
|
for row in reader:
|
|
table.append(row)
|
|
|
|
count += 1
|
|
if count == 100:
|
|
break
|
|
|
|
# Only read first 100 data rows - [2 points] Q5.a
|
|
...
|
|
|
|
|
|
# Order table by the last column - [3 points] Q5.b
|
|
table.sort(key = lambda x: float(x[2]), reverse = True)
|
|
|
|
|
|
return header, table
|
|
|