Matrix reshaping (2d to 1d, 1d to 2d)

import numpy as np 


def convert_1d_array(arr, x, y):
    return arr.reshape(1,x * y)

def convert_2d_array(arr, x, y):
    return arr.reshape(x,y)

def main():
    a = np.ones*1
    a_1d = convert_1d_array(a, 5, 3)
    a_2d = convert_2d_array(a_1d, 5, 3)
    print(a_1d)
    print(a_2d)
    
    
main()

 

大規模の行列計算をするときに、こういった2次元配列を1次元配列に落とし込んだ方が計算が早く済むことがある。

そのメモ書きである。

*1:5,3