在 for 循环中向我的二维数组添加第三维

Question

我有一个 for 循环，它为循环中的每个条目提供 16 x 8 2D 数组的输出。我想将所有这些 2D 数组沿 z 轴堆叠在 3D 数组中。这样，我可以确定 z 轴上的方差。我尝试了多个命令，例如循环内外的 np.dstack、matrix3D[p,:,:] = ... 和 np.newaxis。然而，我最接近我想要的输出只是重复堆叠在一起的最后一个数组。尺寸也很差。我需要保留原来的 16 x 8 格式。到目前为止，我有点太深了，可以在正确的方向使用一些轻推！

我的代码：

excludedElectrodes = [1,a.numberOfColumnsInArray,a.numberOfElectrodes-a.numberOfColumnsInArray+1,a.numberOfElectrodes]
matrixEA = np.full([a.numberOfRowsInArray, a.numberOfColumnsInArray], np.nan)
for iElectrode in range(a.numberOfElectrodes):
    if a.numberOfDeflectionsPerElectrode[iElectrode] != 0:
        matrixEA[iElectrode // a.numberOfColumnsInArray][iElectrode % a.numberOfColumnsInArray] = 0

for iElectrode in range (a.numberOfElectrodes):
    if iElectrode+1 not in excludedElectrodes:
        
        """Preprocessing"""
        
        # Loop over heartbeats
        for p in range (1,len(iLAT)):
            # Calculate parameters, store them in right row-col combo (electrode number)
                    matrixEA[iElectrode // a.numberOfColumnsInArray][iElectrode % a.numberOfColumnsInArray] =  (np.trapz(abs(correctedElectrogram[limitA[0]:limitB[0]]-totalBaseline[limitA[0]:limitB[0]]))/(1000))

                    # Stack all matrixEA arrays along z axis
                    matrix3D = np.dstack(matrixEA)

Answer 1

这个例子 sn-p 做你想做的事，虽然我怀疑你的错误更多地与concatenate 部分无关的事情。在这里，我们在数组中使用None 关键字来创建一个新的空维度（沿此我们连接二维数组）。

import numpy as np

# Function does create a dummy (16,8) array
def foo(a):
    return np.random.random((16,8)) + a

arrays2D = []
# Your loop
for i in range(10):
    # Calculate your (16,8) array
    f = foo(i)
    # And append it to the list
    arrays2D.append(f)

# Stack arrays along new dimension
array3D = np.concatenate([i[...,None] for i in arrays2D], axis = -1)