so_magic.data.backend.panda_handling.client_code

Complexity

Total Complexity

10

Size/Duplication

Total Lines	68
Duplicated Lines	0 %

Importance

Changes

0

10 Methods

Rating	Name	Duplication	Size	Complexity
A	PDTabularRetrieverDelegate.nb_rows()	0	3	1
A	PDTabularRetrieverDelegate.nb_columns()	0	3	1
A	PDTabularIteratorDelegate.itercolumns()	0	3	1
A	PDTabularMutatorDelegate.add_columns()	0	4	1
A	PDTabularIteratorDelegate.columnnames()	0	3	1
A	PDTabularIteratorDelegate.iterrows()	0	3	1
A	PDTabularRetrieverDelegate.row()	0	3	1
A	PDTabularRetrieverDelegate.column()	0	3	1
A	PDTabularMutatorDelegate.add_column()	0	3	1
A	PDTabularRetrieverDelegate.get_numerical_attributes()	0	3	1

import pandas as pd
from so_magic.data.interfaces import TabularRetriever, TabularIterator, TabularMutator

__all__ = ['BACKEND']


# User defined (engine dependent implementations of tabular operations)

class PDTabularRetrieverDelegate(TabularRetriever):
    """The observation object is the same as the one your return from 'from_json_lines'"""

    @classmethod
    def column(cls, identifier, data):
        return data.observations[identifier]

    @classmethod
    def row(cls, identifier, data):
        return data.observations.iloc[identifier]

    @classmethod
    def nb_columns(cls, data):
        return len(data.observations.columns)

    @classmethod
    def nb_rows(cls, data):
        return len(data.observations)

    @classmethod
    def get_numerical_attributes(cls, data):
        return data.observations._get_numeric_data().columns.values


class PDTabularIteratorDelegate(TabularIterator):
    """The observation object is the same as the one your return from 'from_json_lines'"""

    @classmethod
    def columnnames(cls, data):
        return list(data.observations.columns)

    @classmethod
    def iterrows(cls, data):
        return iter(data.observations.iterrows())

    @classmethod
    def itercolumns(cls, data):
        return iter(data.observations[column] for column in data.observations.columns)


class PDTabularMutatorDelegate(TabularMutator):

    @classmethod
    def add_column(cls, datapoints, values, new_attribute, **kwargs):
        datapoints.observations[new_attribute] = values

    @classmethod
    def add_columns(cls, datapoints, values, column_names, **kwargs):
        datapoints.observations = pd.concat([datapoints.observations, pd.DataFrame(data=values, columns=column_names)],
                                            axis=1)


BACKEND = {
    'backend_id': 'pd',
    'backend_name': 'pandas',
    'interfaces': [
        PDTabularRetrieverDelegate,
        PDTabularIteratorDelegate,
        PDTabularMutatorDelegate,
    ]
}