2ba94913fb
new file: aggregates/a-b project/code.py new file: aggregates/codecademy_code/script.py new file: aggregates/codecademy_code/test1.py modified: aggregates/orders.csv new file: aggregates/python-aggregatesI.py new file: aggregates/python-aggregatesII.py new file: aggregates/python-aggregatesIII.py new file: aggregates/python-aggregatesIV.py new file: aggregates/python-aggregatesV.py Added aggregate files
18 lines
693 B
Python
18 lines
693 B
Python
load_file_in_context('script.py')
|
|
|
|
try:
|
|
shoe_counts_pivot
|
|
except NameError:
|
|
fail_tests('Did you remember to define `shoe_counts_pivot`?')
|
|
|
|
ans = orders.groupby(['shoe_type', 'shoe_color']).id.count().reset_index()\
|
|
.pivot(columns='shoe_color', index='shoe_type', values='id')\
|
|
.reset_index()
|
|
|
|
if (set(ans.columns) - set(shoe_counts_pivot.columns)) == set(['shoe_type']):
|
|
fail_tests('Did you remember to use `reset_index()` after `pivot`?')
|
|
elif set(shoe_counts_pivot.columns) != set(ans.columns):
|
|
fail_tests('It looks like you have the wrong columns. You should have {}, but you have {}.'.format(
|
|
', '.join(ans.columns), ', '.join(shoe_counts_pivot.columns)))
|
|
|
|
pass_tests() |