Skip to content
Snippets Groups Projects
Commit f195cfc2 authored by Millian Poquet's avatar Millian Poquet
Browse files

realist: add read_parse_csv

parent d7c129d4
Branches
No related tags found
No related merge requests found
......@@ -65,3 +65,13 @@ def read_parse_several_xls(xls_filenames: [str], lower: bool=None) -> [StudentEn
def student_entry_list_to_df(students: [StudentEntry]) -> pandas.DataFrame:
'''Create a DataFrame from a student list.'''
return pandas.DataFrame(students, columns=student_columns)
def read_parse_csv(csv_filename: str) -> pandas.DataFrame:
df = pandas.read_csv(csv_filename)
expected_columns = set(student_columns)
parsed_columns = {str(x) for x in df.columns}
if not expected_columns.issubset(parsed_columns):
raise RuntimeError(f"missing columns in csv file '{csv_filename}': {expected_columns - parsed_columns}")
return df
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment