Python is_column_subset Examples

Example #1

File: test_exceptions.py Project: vjb/reimagined-octo-fortnight

 def test_is_column_subset(self): 
     
      
     # columns_labels_missing contains column names that aren't in df1, so this should raise an exception
     self.assertRaises(MissingColumns, is_column_subset, self.column_labels_missing, self.df1.columns.tolist())
     
     # column_labels contains column names that are all in df1, so is_column_subset should not raise an exception!
     try:
         is_column_subset(self.column_labels, self.df1.columns.tolist())   
     except MissingColumns:
         self.fail("is_column_subset() raised MissingColumns unexpectedly!")       

Example #2

File: column_extractor.py Project: vjb/reimagined-octo-fortnight

 def transform(self, df, **transform_params):
     
     # sanity checks
     
     is_dataframe(df)
     
     is_column_subset( self.col_labels, df.columns.tolist() )
     
     # extract columns by label
                                
     return df[self.col_labels]