data_fetcher.py 文件源码

def parseVoyagerData(self, spacecraft, in_filename):
        '''
        Parse Voyager Data

        @param spacecraft: Voyager spacecraft (vy1 or vy2)
        @param in_filename: Input voyager data filename

        @return Pandas Dataframe of Voyager data
        '''

        def convert_date(year, day, hour):
            '''
            Convert to datetime

            @param year: Input year
            @param day: Input day
            @param hour: Input hour

            @return datetime
            '''

            return pd.to_datetime("{0:0>4}{1:0>3}{2:0>2}".format(year,day,hour), format='%Y%j%H')


        # Voyager 1 has 3 less columns than Voyager 2
        if spacecraft == 'voyager1':
            field_widths = self.field_widths[:34]
            field_names = self.field_names[:34]
        else:
            field_widths = self.field_widths
            field_names = self.field_names

        # Parse the data
        data = pd.read_fwf(in_filename, widths=field_widths, header=None, names=field_names)

        # Create date column
        data['Date'] = list(map(convert_date,
                                data.loc[:,'Year'],
                                data.loc[:,'Day'],
                                data.loc[:,'Hour']))

        data.set_index('Date', inplace=True)

        return data