2017-09-26 4 views
0

for 루프를 사용하여 numpy 배열 형식을 2 차원 numpy 배열로 만들고 싶습니다. 여러 가지 방법으로 2 차원에서 3 차원 배열을 만들려고 시도했지만 그 각각의 시간은 나 오류. 이것이 내가 한 일이며, 최종 배열의 크기는 (10,3,3)이어야합니다.for 루프를 사용하여 두 번째 numpy 배열에서 동적으로 3d numpy 배열을 만드는 방법

#this is a sample code 
arr=[] 
for i in range(10): 
    a=np.random.rand(3,3) 
    arr=np.stack(a,arr) 
    #arr=np.append(arr,a) 
    #arr=np.array([arr,a]) 
    #arr[i]=a 

답변

1

당신은 list.append 방법을 사용하여 arr 목록에 2d 배열을 추가, 당신은 for 루프로 수행 한 후, np.array로 포장하여 3d 배열에 arr을 변환 할 수 있습니다

arr = [] 
for i in range(10): 
    a = np.random.rand(3,3) 
    arr.append(a) 

np.array(arr).shape 
# (10, 3, 3) 

또는 numpy.stack :

np.stack(arr).shape 
# (10, 3, 3) 
관련 문제