Why Gemfury? Push, build, and install  RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

Repository URL to install this package:

Details    
xlwings / conversion / numpy_conv.py
Size: Mime:
try:
    import numpy as np
except ImportError:
    np = None


if np:

    try:
        import pandas as pd
    except ImportError:
        pd = None

    from . import Converter, Options

    class NumpyArrayConverter(Converter):

        writes_types = np.ndarray

        @classmethod
        def base_reader(cls, options):
            return super(NumpyArrayConverter, cls).base_reader(
                Options(options).defaults(empty=np.nan)
            )

        @classmethod
        def read_value(cls, value, options):
            dtype = options.get("dtype", None)
            copy = options.get("copy", True)
            order = options.get("order", None)
            ndim = options.get("ndim", None) or 0
            return np.array(value, dtype=dtype, copy=copy, order=order, ndmin=ndim)

        @classmethod
        def write_value(cls, value, options):
            return value.tolist()

    NumpyArrayConverter.register(np.array, np.ndarray)