def partition_df(df, col): # return dictionary object of filtered data frame for each unique element in col
    try:
        # dict comprehension creates dict to holds DataFrame for each country
        dict1 = {k: df[df[col] == k] for k in m.get_unique_vals(df, col)}
        return dict1
    except (TypeError):
         print(TypeError('arg1 must be a DataFrame'))
    except KeyError as e:
        cause = e.args[0]
        print(cause + ' not a valid column in the dataframe')
def partition_df(
    df, col
):  # return dictionary object of filtered data frame for each unique element in col
    try:
        # dict comprehension creates dict to holds DataFrame for each country
        dict1 = {k: df[df[col] == k] for k in m.get_unique_vals(df, col)}
        return dict1
    except (TypeError):
        print(TypeError('arg1 must be a DataFrame'))
    except KeyError as e:
        cause = e.args[0]
        print(cause + ' not a valid column in the dataframe')
Ejemplo n.º 3
0
 def test_returns_only_unique_vals(self):
     self.assertEquals(len(v.filter_by_col_val(df, 'pop', 0)), 2)
Ejemplo n.º 4
0
 def test_if_col_not_exist_keyerror(self):
     self.assertRaises(KeyError,
                       v.filter_by_col_val(df, 'qwerty1', 'qwerty2'))
Ejemplo n.º 5
0
 def test_amount_needed_calculation_correct(self):
     new_frame = v.add_amount_needed(df, 'goal', 'raised')
     amount_n = new_frame['amount_needed']
     self.assertEquals(amount_n[0], 2)
     self.assertEquals(amount_n[5], 0)
Ejemplo n.º 6
0
 def test_if_col_not_exist_keyerror(self):
     self.assertRaises(KeyError,
                       v.add_amount_needed(df, 'qwerty1', 'qwerty2'))
Ejemplo n.º 7
0
 def test_returns_only_unique_vals(self):
     self.assertEquals(len(v.get_unique_vals(df, 'country')), 5)
     self.assertEquals(len(v.get_unique_vals(df, 'pop')), 2)
Ejemplo n.º 8
0
 def test_if_col_not_exist_keyerror(self):
     self.assertRaises(KeyError, v.get_unique_vals(df, 'qwerty1'))
 def test_returns_only_unique_vals(self):
     self.assertEquals(len(v.filter_by_col_val(df, 'pop', 0)), 2)
 def test_if_col_not_exist_keyerror(self):
     self.assertRaises(KeyError, v.filter_by_col_val(df, 'qwerty1', 'qwerty2'))
 def test_amount_needed_calculation_correct(self):
     new_frame = v.add_amount_needed(df, 'goal', 'raised')
     amount_n = new_frame['amount_needed']
     self.assertEquals(amount_n[0], 2)
     self.assertEquals(amount_n[5], 0)
 def test_if_col_not_exist_keyerror(self):
     self.assertRaises(KeyError, v.add_amount_needed(df, 'qwerty1', 'qwerty2'))
 def test_returns_only_unique_vals(self):
     self.assertEquals(len(v.get_unique_vals(df, 'country')), 5)
     self.assertEquals(len(v.get_unique_vals(df, 'pop')), 2)
 def test_if_col_not_exist_keyerror(self):
     self.assertRaises(KeyError, v.get_unique_vals(df, 'qwerty1'))