如何连接向量以在循环中创建数组?

问题描述 投票:1回答:1

我有一个循环for,在每次迭代中生成一个新的向量(100,)。所以代码循环喜欢

for i in range (10):
   for j in range (4):
     #Create a new vector (100,) 
     #Concatenate 4 vector together to make (400,) #400=4*10 
#Append the concatenation vectors (400,) in vertical to make (10,400) array

我的预期是生成一个(10,400)的矩阵大小,在这些循环中连接向量

目前,我的解决方案是

   matrix_= np.empty([10,400])
    for i in range (10):
       vector_horz=[]
       for j in range (4):
         #Create a new vector (100,) 
         vector_rnd=#Random make a vector/list with size of (100,1)
         #Concatenate 4 vector together to make (400,) #400=4*10
         vector_horz.append(vector_rnd)
    #Append the concatenation vectors (400,) in vertical to make (10,400) array
       matrix_(:,i)=vector_horz

但是,它说我的matrixvector_horz的大小不能分配。你能给我另一种解决方案吗?

python arrays numpy
1个回答
3
投票

选项1 (推荐)首先生成数据并在最后创建一个数组:

data = []
for i in range(10):
    for j in range(4):
        temp_list = ... # random vector of 100 elements 
        data.extend(temp_list) 

arr = np.reshape(data, (10, 400))

选项2 或者,使用np.empty初始化一个空数组,并一次分配一个切片:

arr = np.empty((10, 400))
for i in range(10):
    for j in range(4):
        temp_list = ...
        arr[i, j * 100 : (j + 1) * 100] = temp_list
© www.soinside.com 2019 - 2024. All rights reserved.