def clean_bundles(bundles, length_thr): new_bundles = [] for bundle in bundles: lengths = length_distribution(bundle) new_bundle = [s for (i, s) in enumerate( bundle) if lengths[i] > length_thr] new_bundles.append(new_bundle) return new_bundles
def all_descriptors(bundles): descr = {} descr['lengths'] = [] descr['avg_streamline'] = [] descr['winding_angle'] = [] descr['dragons_hits'] = [] for bundle in bundles: bundle = flip_to_source(bundle, bundle[0][0]) descr['lengths'].append(length_distribution(bundle)) avg = avg_streamline(bundle) descr['avg_streamline'].append(avg) descr['winding_angle'].append(winding_angles(bundle)) descr['dragons_hits'].append(dragons_hits(bundle, avg)) return descr