previous.py 文件源码

def get_data(self, dataset, event_list=None):
        # Load Basics for this dataset and shift it by 1
        data = hax.minitrees.load_single_minitree(dataset, 'Basics')
        df = data.shift(1)

        # Add previous_ prefix to all columns
        df = df.rename(columns=lambda x: 'previous_' + x)

        # Add (unshifted) event number and run number, to support merging
        df['event_number'] = data['event_number']
        df['run_number'] = data['run_number']

        # Support for event list (lame)
        if event_list is not None:
            df = df[np.in1d(df['event_number'].values, event_list)]

        return df