def reorder_dataframe_dataset()

in spotify_tensorflow/featran.py [0:0]

• 7 lines of code
• 2 McCabe index (conditional complexity)

    def reorder_dataframe_dataset(cls,
                                  dataset,  # type: Iterator[pd.DataFrame]
                                  settings_path  # type: str
                                  ):
        # type: (...) -> Iterator[pd.DataFrame]
        """
        Reorders a pandas DataFrame so that feature columns are in the same order as those in a
        Featran settings file.

        :param dataset: A dataset created via Datasets.dataframe
        :param settings_path: Path to the Featran Settings JSON Directory
        :return: An iterator over new DataFrame batches with ordered columns
        """
        feature_names = Featran.names(settings_path)
        for batch in dataset:
            yield batch[feature_names]