나는야 데이터사이언티스트/PYTHON

[Python]3차원 array를 DataFrame으로 만들기(3d arrays to DataFrame)

우주먼지의하루 2021. 11. 13. 00:02
728x90

DataFrame으로 안만들면 안되는 병이 있음...

딥러닝 아직 잘 몰라서 Array가 익숙하지 않음..

 

난 정리된 테이블 형태가 좋다고 !!!

 

어쨌든... 3차원 array를 DataFrame으로 만드는 방법 꼬우 !

 

 

내 array 상태는 이렇게 생김

y_pred.shape는 (229445, 1, 6)

아래 코드를 쓰면 DataFrame으로 만들어짐

#3차원 array를 DataFrame으로 만드는 방법

m,n,r = y_pred.shape
out_arr = np.column_stack((np.repeat(np.arange(m),n),y_pred.reshape(m*n,-1)))
out_df = pd.DataFrame(out_arr)

 

혹시 그냥 1행으로 만들고 싶다면...!

#3차원 array를 DataFrame으로 만드는 방법
pd.DataFrame(dict(rows = list(y_pred)))

 

반응형