Source code for bronx.syntax.arrays

# -*- coding: utf-8 -*-

"""
Numpy's useful tools.
"""

from __future__ import print_function, absolute_import, unicode_literals, division

import numpy


#: No automatic export
__all__ = []


def stretch_array(array):
    """
    Return array.flatten() or compressed(), whether the array is
    masked or not.
    """
    if isinstance(array, numpy.ma.masked_array):
        array = array.compressed()
    elif isinstance(array, numpy.ndarray):
        array = array.flatten()
    else:
        raise NotImplementedError('type: {} array'.format(type(array)))
    return array