14

首先,这是代码的相关部分:

stokes_list = np.zeros(shape=(numrows,1024)) # 'numrows' defined earlier
for i in range(numrows):
    epoch_name = y['filename'][i] # 'y' is an array from earlier
    os.system('pdv -t {0} > temp.txt '.format(epoch_name)) # 'pdv' is a command from another piece of software - here I copy the output into a temporary file
    stokes_line = np.genfromtxt('temp.txt', usecols=3, dtype=[('stokesI','float')], skip_header=1)
    stokes_list = np.vstack((stokes_line,stokes_line))

因此,基本上,每次代码循环时,都会stokes_line从文件中拉出一列(第 4 列)temp.txt,并且我希望它stokes_list每次都添加一行。

例如,如果第一个stokes_line

1.1 2.2 3.3  

第二个是

4.4 5.5 6.6  

那么stokes_list将是

1.1 2.2 3.3  
4.4 5.5 6.6  

并且会继续增长...

目前它不起作用,因为我认为这条线:

stokes_list = np.vstack((stokes_line,stokes_line))

是不正确的。它只是堆叠 2 个列表 - 这是有道理的,因为我只有 2 个参数。我基本上想知道我如何一次又一次地堆叠。

任何帮助将不胜感激!
如果需要,这里是 temp.txt 文件格式的示例:

File: t091110_065921.SFTC Src: J1903+0925 Nsub: 1 Nch: 1 Npol: 4 Nbin: 1024 RMS: 0.00118753  
0 0 0 0.00148099 -0.00143755 0.000931365 -0.00296775  
0 0 1 0.000647476 -0.000896698 0.000171287 0.00218597  
0 0 2 0.000704697 -0.00052846 -0.000603842 -0.000868739  
0 0 3 0.000773361 -0.00234724 -0.0004112 0.00358033  
0 0 4 0.00101559 -0.000691062 0.000196023 -0.000163109  
0 0 5 -0.000220367 -0.000944024 0.000181002 -0.00268215  
0 0 6 0.000311783 0.00191545 -0.00143816 -0.00213856  
4

2 回答 2

42

vstack一遍又一遍地 ing 不好,因为它复制了整个数组。

创建一个普通的 Python list.append然后将其全部传递给它np.vstack以创建一个新数组。

stokes_list = []
for i in xrange(numrows):
    ...
    stokes_line = ...
    stokes_list.append(stokes_line)

big_stokes = np.vstack(stokes_list)
于 2012-09-06T11:45:36.550 回答
11

stokes_list您已经知道数组的最终大小,因为您知道numrows. 所以看起来你不需要增长一个数组(这是非常低效的)。您可以在每次迭代中简单地分配正确的行。只需将最后一行替换为:

stokes_list[i] = stokes_line

顺便说一句,关于您的非工作线,我认为您的意思是:

stokes_list = np.vstack((stokes_list, stokes_line))

您正在用stokes_list它的新值替换的地方。

于 2012-09-06T11:47:59.840 回答