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)