#!python
from pyPRISM.core.MatrixArray import MatrixArray
from pyPRISM.core.Space import Space
import numpy as np
[docs]class IdentityMatrixArray(MatrixArray):
'''Specialization of MatrixArray which is initialized with Identity matrices
See :class:`pyPRISM.core.MatrixArray` for details
'''
def __init__(self,length,rank,data=None,space=None,types=None):
super(IdentityMatrixArray,self).__init__(length=length,rank=rank,space=space,types=types)
self.data = np.zeros((length,rank,rank))
for i in range(rank):
self.data[:,i,i] = 1.0