Python pandas的DataFrame和numpy的Array互相转换

一、DataFrame转换为Array

>>> import pandas as pd
>>> import numpy as np
>>> 
>>> df = pd.DataFrame(np.arange(12).reshape(3, 4), index = ['row1', 'row2', 'row3'], columns=['col1', 'col2', 'col3', 'col4'])
>>> 
>>> df
      col1  col2  col3  col4
row1     0     1     2     3
row2     4     5     6     7
row3     8     9    10    11
>>> 
>>> array = np.array(df)
>>> 
>>> array
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

二、Array转换为DataFrame

>>> import pandas as pd
>>> import numpy as np
>>> 
>>> df = pd.DataFrame(np.arange(12).reshape(3, 4), index = ['row1', 'row2', 'row3'], columns=['col1', 'col2', 'col3', 'col4'])
>>> 
>>> df
      col1  col2  col3  col4
row1     0     1     2     3
row2     4     5     6     7
row3     8     9    10    11
>>> 
>>> array = np.array(df)
>>> 
>>> array
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])
>>> 
>>> 
>>> df1 = pd.DataFrame(array)
>>> 
>>> df1
   0  1   2   3
0  0  1   2   3
1  4  5   6   7
2  8  9  10  11
>>> 
>>> 
>>> 
>>> df1 = pd.DataFrame(array, index = ['row1', 'row2', 'row3'], columns=['col1', 'col2', 'col3', 'col4'])
>>> 
>>> df1
      col1  col2  col3  col4
row1     0     1     2     3
row2     4     5     6     7
row3     8     9    10    11

Array转换为DataFrame可以用index和columns指定行名和列名。

引用

【1】https://pandas.pydata.org/pandas-docs/stable/reference/api/pandas.DataFrame.html