iframe-proxy | Sunbelt Computer Software

53 lines (43 loc) · 1.37 KB

Utility functions for working with and reasoning about arrays.
from typing import Any
def is_arraylike(arr: Any) -> bool:
    Return True iff the object is arraylike: possessing
    .shape, .dtype, .__array__, and .ndim attributes.
    :param arr: The object to check for arraylike properties
    :return: True iff the object is arraylike
    return (
        hasattr(arr, "shape")
        and hasattr(arr, "dtype")
        and hasattr(arr, "__array__")
        and hasattr(arr, "ndim")
def is_memoryarraylike(arr: Any) -> bool:
    Return True iff the object is memoryarraylike:
    an arraylike object whose .data type is memoryview.
    :param arr: The object to check for memoryarraylike properties
    :return: True iff the object is memoryarraylike
    return (
        is_arraylike(arr)
        and hasattr(arr, "data")
        and type(arr.data).__name__ == "memoryview"
def is_xarraylike(xarr: Any) -> bool:
    Return True iff the object is xarraylike:
    possessing .values, .dims, and .coords attributes,
    and whose .values are arraylike.
    :param arr: The object to check for xarraylike properties
    :return: True iff the object is xarraylike
    return (
        hasattr(xarr, "values")
        and hasattr(xarr, "dims")
        and hasattr(xarr, "coords")
        and is_arraylike(xarr.values)

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

_arrays.py

Latest commit

History

_arrays.py

File metadata and controls