Source code for draugr.numpy_utilities.manipulation

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "Christian Heider Nielsen"
__doc__ = r"""

           Created on 17/07/2020
           """

__all__ = ["recursive_flatten_numpy", "MyNdArray"]

from typing import Sequence

import numpy


[docs]def recursive_flatten_numpy(nested: Sequence) -> numpy.ndarray: """description""" return numpy.array(nested).ravel()
[docs]class MyNdArray(numpy.ndarray): """ Small extension """ @property def conjugate_transpose(self): """description :return: """ return self.conj().T
if __name__ == "__main__": print(recursive_flatten_numpy([[numpy.zeros((2, 2))], [numpy.zeros((2, 2))]])) print(recursive_flatten_numpy(numpy.zeros((2, 2)))) print(numpy.random.random((3, 3)).view(MyNdArray).conjugate_transpose)