molecule_2state_wigner_moyal.py 文件源码

def theta_slice(self, *args):
        """
        Slice array A listed in args as if they were returned by fft.rfft(A, axis=0)
        """
        return (A[:(1 + self.P_gridDIM//2), :] for A in args)