2

我有一个函数，比如 peaksdetect()，它将生成一个未知行数的二维数组；我会叫它几次，比如说 3，我想用这 3 个数组，一个 3-D 数组。这是我的开始，但是有很多 if 语句非常复杂，所以如果可能的话，我想让事情变得更简单：

import numpy as np

dim3 = 3   # the number of times peaksdetect() will be called
           # it is named dim3 because this number will determine 
           # the size of the third dimension of the result 3-D array

for num in range(dim3):
    data = peaksdetect(dataset[num])            # generates a 2-D array of unknown number of rows
    if num == 0:
        3Darray = np.zeros([dim3, data.shape])  # in fact the new dimension is in position 0
                                                # so dimensions 0 and 1 of "data" will be 
                                                # 1 and 2 respectively
    else:
        if data.shape[0] > 3Darray.shape[1]:
            "adjust 3Darray.shape[1] so that it equals data[0] by filling with zeroes"
            3Darray[num] = data
        else:
            "adjust data[0] so that it equals 3Darray.shape[1] by filling with zeroes"
            3Darray[num] = data
...
4

1 回答 1

2

如果您指望必须调整数组的大小，那么通过预分配它很可能不会获得太多收益。将数组存储在一个列表中可能会更简单，然后计算出数组的大小来保存它们，然后将数据转储到其中：

data = []
for num in range(dim3):
    data.append(peaksdetect(dataset[num]))
shape = map(max, zip(*(j.shape for j in data)))
shape = (dim3,) + tuple(shape)
data_array = np.zeros(shape, dtype=data[0].dtype)
for j, d in enumerate(data):
    data_array[j, :d.shape[0], :d.shape[1]] = d
于 2013-04-20T20:39:30.003 回答