B d @sfddlZddlZddlmZddlmZddlm Z ddl m Z dgZ ddZ ddddd d dZdS) N)units)AstropyUserWarning) TimeSeries)BinnedTimeSeriesaggregate_downsamplec Cst|drt|||Sg}xjtt|dD]V}||d||dkrf|||||q2|||||||dq2W||||ddt|SdS)z Manual reduceat functionality for cases where Numpy functions don't have a reduceat. It will check if the input function has a reduceat and call that if it does. reduceatN)hasattrnparrayrrangelenappend)r indicesfunctionresultirj/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/astropy/timeseries/downsample.pyrs &r) time_bin_sizetime_bin_startn_binsaggregate_funcc Cst|tstdt|tjs&td|tj}|jdd}|dkrR|jd}|j|j }|dkr|t t |d|}|dkrt j }t t dt ||g}||tj} ||dk||dk@} || } t ||| d} d| || |dk<t| dd| dd} t dt t | ddg}t | }x| jD]}|dkrZqH| |}t|t jtjfstd tqHt|tjrtjt t j||jd }tjt|j|||jd d ||<n2t jj ||j!d }d|_"t|||||<d|j"|<|| |<qHW| S)a Downsample a time series by binning values into bins with a fixed size, using a single function to combine the values in the bin. Parameters ---------- time_series : :class:`~astropy.timeseries.TimeSeries` The time series to downsample. time_bin_size : `~astropy.units.Quantity` ['time'] The time interval for the binned time series. time_bin_start : `~astropy.time.Time`, optional The start time for the binned time series. Defaults to the first time in the sampled time series. n_bins : int, optional The number of bins to use. Defaults to the number needed to fit all the original points. aggregate_func : callable, optional The function to use for combining points in the same bin. Defaults to np.nanmean. Returns ------- binned_time_series : :class:`~astropy.timeseries.BinnedTimeSeries` The downsampled time series. z"time_series should be a TimeSeriesz/time_bin_size should be a astropy.unit quantityNrr r)rZ time_bin_endtimez.Skipping column {0} since it has a mix-in type)unitF)copy)dtype)# isinstancer TypeErroruZQuantityZto_valuesZilocrsecintr ceilZnanmeanZcumsumZhstackrepeatZ searchsortedrZnonzerodiffuniqueZcolnamesZndarraywarningswarnrnanrrvaluemaZzerosrmask)Z time_seriesrrrrZ bin_size_secsortedZrelative_time_secZrelative_bins_secZbinsZkeepZsubsetrZbinnedgroupsZunique_indicesZcolnamevaluesdatarrrr!sP      "     )r(numpyr Zastropyrr Zastropy.utils.exceptionsrZastropy.timeseries.sampledrZastropy.timeseries.binnedr__all__rrrrrrs