[파이썬] 문제 : numpy로 a 행렬을 b와 같은 형태로 만들기(concatenate, stack)

 

import numpy as np

a = [[0,1,2], [3,4,5], [0,2,4],[6,8,10]]

b1 = np.concatenate((np.array(a[0]), np.array(a[2])), axis=0)   # [0,1,2] + [0,2,4] = [0,1,2,0,2,4]
b2 = np.concatenate((np.array(a[1]), np.array(a[3])), axis=0)   # [3,4,5] + [6,8,10] = [3,4,5,6,8,10]
b = np.stack((b1, b2), axis=0)                                  # [[0,1,2,0,2,4], [3,4,5,6,8,10]]
print(b)

결과

[[ 0  1  2  0  2  4]
 [ 3  4  5  6  8 10]]