diff --git a/nifty5/multi/multi_field.py b/nifty5/multi/multi_field.py index cce6c79a1171986071caee5a1ef88c969a45d49d..b878cf84e000b27c0daba36dbd251fb7a49fea8b 100644 --- a/nifty5/multi/multi_field.py +++ b/nifty5/multi/multi_field.py @@ -52,6 +52,16 @@ class MultiField(object): def dtype(self): return {key: val.dtype for key, val in self._val.items()} + @property + def real(self): + """MultiField : The real part of the multi field""" + return MultiField({key: field.real for key, field in self.items()}) + + @property + def imag(self): + """MultiField : The imaginary part of the multi field""" + return MultiField({key: field.imag for key, field in self.items()}) + @staticmethod def from_random(random_type, domain, dtype=np.float64, **kwargs): dtype = MultiField.build_dtype(dtype, domain)