Question

How to flatten a 2D matrix to 1D array in column order with Python

Answer 1

import numpy as np

arr = np.array([[1, 2, 3], 
                [4, 5, 6], 
                [7, 8, 9]])

arr = np.ravel(arr.T)

print(arr)





'''
run:

[1 4 7 2 5 8 3 6 9]

'''

 

70+ SQL courses for beginners and professionals

Answer 2

import numpy as np

arr = np.array([[1, 2, 3], 
                [4, 5, 6], 
                [7, 8, 9]])

arr = np.ravel(arr, order='F')

print(arr)





'''
run:

[1 4 7 2 5 8 3 6 9]

'''

 

70+ SQL courses for beginners and professionals