0

我正在编写一个程序来从列表中读取数据，傅里叶变换并在绘图之前对其进行移位。到目前为止，代码从 DICOM 文件中获取光谱数据并将其放入列表中，每个元素都是一个包含每个 FID/光谱值的数组。

from pylab import *
import dicom

plan=dicom.read_file("")

all_points = array(plan.SpectroscopyData)
cmplx_data = all_points[0::2] + 1j*all_points[1::2]
frames = int(plan.NumberOfFrames)
fid_pts = len(cmplx_data)/frames

fid_list = []
for fidN in arange(frames):
    offset = fidN * fid_pts 
    current_fid = cmplx_data[offset:offset+fid_pts]
    fid_list.append(current_fid)

这可以很好地对数据进行分组，但是在尝试使用生成的数组时遇到了问题。首先，当试图只显示数据的复杂部分时，例如：

plot(complex(fid_list[0]))

退货 

Traceback (most recent call last)
/home/dominicc/Desktop/<ipython-input-37-4146b7fbfd7c> in <module>()
----> 1 plot(complex(fid_list[0]))

TypeError: only length-1 arrays can be converted to Python scalars

其次，也是最重要的是，在尝试绘制 FFT 数据的零频移时，我遇到了无限递归：

plot(fftshift(fft(fid_list[0])))

收到以下错误

/home/dominicc/Desktop/New_Script.py in fftshift(fid_in)
     23 
     24 def fftshift(fid_in):
---> 25         fft_fid_in = fft(fid_in)
     26         plot(fftshift(fft_fid_in))
     27         show()

/usr/lib/python2.7/dist-packages/numpy/fft/fftpack.pyc in fft(a, n, axis)
    162     """
    163 
--> 164     return _raw_fft(a, n, axis, fftpack.cffti, fftpack.cfftf, _fft_cache)
    165 
    166 

/usr/lib/python2.7/dist-packages/numpy/fft/fftpack.pyc in _raw_fft(a, n, axis, init_function, work_function, fft_cache)
     43 def _raw_fft(a, n=None, axis=-1, init_function=fftpack.cffti,
     44              work_function=fftpack.cfftf, fft_cache = _fft_cache ):
---> 45     a = asarray(a)
     46 
     47     if n is None:

RuntimeError: maximum recursion depth exceeded

任何人都可以提出改进我的代码以避免这些问题的方法吗？谢谢。

4

2 回答 2

1

第一个错误

在你的 for 循环中，你有：

current_fid = cmplx_data[offset:offset+fid_pts]
fid_list.append(current_fid)

因此，fid 是一个多维列表。之所以如此，是因为[foo,bar].append([some,list])导致[foo,bar,[some,list]]

complex(fid_list[0])期望它得到的任何列表的长度为 1。该行current_fid = cmplx_data[offset:offset+fid_pts]表示 fid_list[0] 的长度fid_pts

第二个错误

所以递归函数需要在内部处理两个分支。一个是终止分支（这可以阻止事物螺旋式上升到无穷大），另一个分支是尝试上述螺旋式的分支。

fftshift 不负责终止分支，因此您需要为此添加一些代码。

像这样重写函数并运行它以查看这一点的说明：

def fftshift(fid_in):
    print('fftshift 1')
    fft_fid_in = fft(fid_in)
    print('fftshift 2')
    foo = fftshift(fft_fid_in)
    print(' fftshift3')

这将打印：

fftshift 1
fftshift 2
fftshift 1
fftshift 2
fftshift 1
fftshift 2
etc etc recursion error

递归函数的最简单形式是：

def my_recursive_fn(foo):
    if some_condition:   #the terminating condition
        return bar #this should NOT call my_recursive_fn in any way
    moo = do_processing(foo)
    return my_recursive_fn(foo)


    plot(foo)
    show()
于 2012-11-29T14:36:57.300 回答
0

通过在 fftshit 之前添加傅里叶转换数据的另一个 for 循环来解决：

fft_list = []
for i in range(0, frames):
    current_fid = fft(fid_list[i])
    fft_list.append(current_fid)
于 2012-12-03T12:44:35.343 回答