Monday, September 4, 2017

Python: extracting column from np array and making it flat

>>> import numpy as np
>>> A = np.array([[1,2,3,4],[5,6,7,8]])

>>> A
array([[1, 2, 3, 4],
    [5, 6, 7, 8]])

>>> A[:,2] # returns the third columm
>>> import pandas as pd
>>>aa = pd.DataFrame(A[:,2].ravel())
>>>aa

matlab mat file in python

import scipy.io as sio
test = sio.loadmat('test.mat')
test
{'a': array([[[  1.,   4.,   7.,  10.],
        [  2.,   5.,   8.,  11.],
        [  3.,   6.,   9.,  12.]]]),
 '__version__': '1.0',
 '__header__': 'MATLAB 5.0 MAT-file, written by
 Octave 3.6.3, 2013-02-17 21:02:11 UTC',
 '__globals__': []}
>>> oct_a = test['a']
>>> oct_a