B d==@s~ddlmZddlZddlmZmZmZddlm Z m Z ddl m Z ddlmZddlmZmZdgZeGd ddeZdS) )deepcopyN)groupsTableQTable)Time TimeDelta)units)Quantity)BaseTimeSeriesautocheck_required_columnsBinnedTimeSeriescseZdZdZddgZddddddfdd Zedd Zed d Zed d Z eddZ fddZ e dfdd Z ZS)r a A class to represent binned time series data in tabular form. `~astropy.timeseries.BinnedTimeSeries` provides a class for representing time series as a collection of values of different quantities measured in time bins (for time series with values sampled at specific times, see the `~astropy.timeseries.TimeSeries` class). `~astropy.timeseries.BinnedTimeSeries` is a sub-class of `~astropy.table.QTable` and thus provides all the standard table maniplation methods available to tables, but it also provides additional conveniences for dealing with time series, such as a flexible initializer for setting up the times, and attributes to access the start/center/end time of bins. See also: https://docs.astropy.org/en/stable/timeseries/ Parameters ---------- data : numpy ndarray, dict, list, table-like object, optional Data to initialize time series. This does not need to contain the times, which can be provided separately, but if it does contain the times they should be in columns called ``'time_bin_start'`` and ``'time_bin_size'`` to be automatically recognized. time_bin_start : `~astropy.time.Time` or iterable The times of the start of each bin - this can be either given directly as a `~astropy.time.Time` array or as any iterable that initializes the `~astropy.time.Time` class. If this is given, then the remaining time-related arguments should not be used. This can also be a scalar value if ``time_bin_size`` is provided. time_bin_end : `~astropy.time.Time` or iterable The times of the end of each bin - this can be either given directly as a `~astropy.time.Time` array or as any value or iterable that initializes the `~astropy.time.Time` class. If this is given, then the remaining time-related arguments should not be used. This can only be given if ``time_bin_start`` is an array of values. If ``time_bin_end`` is a scalar, time bins are assumed to be contiguous, such that the end of each bin is the start of the next one, and ``time_bin_end`` gives the end time for the last bin. If ``time_bin_end`` is an array, the time bins do not need to be contiguous. If this argument is provided, ``time_bin_size`` should not be provided. time_bin_size : `~astropy.time.TimeDelta` or `~astropy.units.Quantity` The size of the time bins, either as a scalar value (in which case all time bins will be assumed to have the same duration) or as an array of values (in which case each time bin can have a different duration). If this argument is provided, ``time_bin_end`` should not be provided. n_bins : int The number of time bins for the series. This is only used if both ``time_bin_start`` and ``time_bin_size`` are provided and are scalar values. **kwargs : dict, optional Additional keyword arguments are passed to `~astropy.table.QTable`. time_bin_start time_bin_sizeN)r time_bin_endrn_binsc stjfd|i||dkrH|dkrH|dkrH|dkrH|dkrHd|_dSd|jkrn|dkrf|jd}ntdd|jkr|dkr|jd}ntd|dkrtd|dkr|dkrtdt|ttfst|}|dk rt|ttfst|}|dk rt|t tfstd t|tr*|j t j }|dk rV|dk rV|j rN|j sVtd |j r|dkrptd |j r|dk r|dk r|t|krtd nt|}t||}t|}||}t|d }dt j |d<||}nt|jdkr(t|t|kr(tdt|t||dk rr|j rb|}|d d|dd<||d<|}||j t j }|j rt|t|}|\d|jkr|dd|jkr|d|j|ddd|d|j|d ddWdQRXdS)NdataTr zK'time_bin_start' has been given both in the table and as a keyword argumentrzJ'time_bin_size' has been given both in the table and as a keyword argumentz''time_bin_start' has not been specifiedzLength of 'time_bin_start' ({}) should match table length ({}))indexname)super__init__Z_required_columns_relaxcolnamescolumns TypeError isinstancerrr secusisscalarlennprepeatZcumsumZroll ValueErrorformatcopyZ_delay_required_column_checks remove_column add_columnZ add_index) selfrr rrrkwargsZ time_deltatimes) __class__f/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/astropy/timeseries/binned.pyrJs~              "       zBinnedTimeSeries.__init__cCs|dS)z7 The start times of all the time bins. r r,)r(r,r,r-r szBinnedTimeSeries.time_bin_startcCs|d|ddS)z8 The center times of all the time bins. r rg?r,)r(r,r,r-time_bin_centersz BinnedTimeSeries.time_bin_centercCs|d|dS)z5 The end times of all the time bins. r rr,)r(r,r,r-rszBinnedTimeSeries.time_bin_endcCs|dS)z1 The sizes of all the time bins. rr,)r(r,r,r-rszBinnedTimeSeries.time_bin_sizecsh|r\d|ksd|kr\tfdd|Dtjjd}tj|jjjjd|_ |St |S)Nr rcsg|] }|qSr,r,).0x)r(r,r- sz0BinnedTimeSeries.__getitem__..)metaZ copy_indices)indiceskeys) Z_is_list_or_tuple_of_strrrr2Z _copy_indicesrZ TableGroupsZ_indicesZ_keysZ_groupsr __getitem__)r(itemout)r+)r(r-r5s   zBinnedTimeSeries.__getitem__c  sytj|f| d|i| Stk r|dkr>td|dkrX|dkrXtdn|dk rp|dk rptdtj|f| d|i| } || jkrt| j|||d} | |ntd|d|dk r|| jkrt| j|||d} | |ntd |dd}n|dk r|| jkr:| j|}| |ntd |d|j dkr|dksnt |t j svtd ||}n t |}d} | jr|jrt| | | |t| d St| | | |d SYnXdS)a Read and parse a file and returns a `astropy.timeseries.BinnedTimeSeries`. This method uses the unified I/O infrastructure in Astropy which makes it easy to define readers/writers for various classes (https://docs.astropy.org/en/stable/io/unified.html). By default, this method will try and use readers defined specifically for the `astropy.timeseries.BinnedTimeSeries` class - however, it is also possible to use the ``format`` keyword to specify formats defined for the `astropy.table.Table` class - in this case, you will need to also provide the column names for column containing the start times for the bins, as well as other column names (see the Parameters section below for details):: >>> from astropy.timeseries.binned import BinnedTimeSeries >>> ts = BinnedTimeSeries.read('binned.dat', format='ascii.ecsv', ... time_bin_start_column='date_start', ... time_bin_end_column='date_end') # doctest: +SKIP Parameters ---------- filename : str File to parse. format : str File format specifier. time_bin_start_column : str The name of the column with the start time for each bin. time_bin_end_column : str, optional The name of the column with the end time for each bin. Either this option or ``time_bin_size_column`` should be specified. time_bin_size_column : str, optional The name of the column with the size for each bin. Either this option or ``time_bin_end_column`` should be specified. time_bin_size_unit : `astropy.units.Unit`, optional If ``time_bin_size_column`` is specified but does not have a unit set in the table, you can specify the unit manually. time_format : str, optional The time format for the start and end columns. time_scale : str, optional The time scale for the start and end columns. *args : tuple, optional Positional arguments passed through to the data reader. **kwargs : dict, optional Keyword arguments passed through to the data reader. Returns ------- out : `astropy.timeseries.binned.BinnedTimeSeries` BinnedTimeSeries corresponding to the file. r$Nz\``time_bin_start_column`` should be provided since the default Table readers are being used.zJEither `time_bin_end_column` or `time_bin_size_column` should be provided.zECannot specify both `time_bin_end_column` and `time_bin_size_column`.)scaler$zBin start time column 'z' not found in the input data.zBin end time column 'zBin size column 'zVThe bin size unit should be specified as an astropy Unit using ``time_bin_size_unit``.)rr rrr)rr rr)rreadrr#rrrrr&unitrrZUnitBaser rr r )r(filenameZtime_bin_start_columnZtime_bin_end_columnZtime_bin_size_columnZtime_bin_size_unit time_formatZ time_scaler$argsr)tabler rr)r+r,r-r9sV8                  zBinnedTimeSeries.read)N)NNNNNNN)__name__ __module__ __qualname____doc__Z_required_columnsrpropertyr r.rrr5 classmethodr9 __classcell__r,r,)r+r-r s5p     )r%rnumpyr!Z astropy.tablerrrZ astropy.timerrZastropyrrZ astropy.unitsr Zastropy.timeseries.corer r __all__r r,r,r,r-s