def add_partition_features(prep, grid_length, grid_width): feat_dicts = [] # length of list is number of obs, each element is dictionary of agged feature for i in range(len(prep.raw_data)): mapped_image = prep.get_mapped_image(i) part_dict = PreProcessor.partition_image(mapped_image, grid_length = grid_length, grid_width = grid_width) feat_dicts.append(PreProcessor.agg_partitions(part_dict)) key_list = list(feat_dicts[0].keys()) for key in key_list: prep.add_feature(None, [feat_dicts[i][key] for i in range(len(feat_dicts))])