将三维矩阵转换为垂直级联的一维阵列

2024-04-19 06:18:54 发布

您现在位置:Python中文网/ 问答频道 /正文

下面这段代码中有没有指向for循环的快捷方式?在

import numpy as np

ff = np.zeros((3,3,4))
gg = np.zeros((36,1))

ff[:,:,0] = 1
ff[:,:,1] = 2
ff[:,:,2] = 3
ff[:,:,3] = 4


for ii in xrange(4):
    gg[ii*9:(ii+1)*9,0] = ff[:,:,ii].reshape((9,))

Tags: 代码inimportnumpyforasnpzeros