【Python】数据框DataFrame和列表List相互转换

0x00 前言

在使用一些别人封装好的库的时候，调用函数返回的结果便是DataFrame，这时如果要对内部数据做一些加工处理的话会很不方便。我们要需要将DataFrame还原成列表的形式来处理。

0x01 DataFrame → List

import numpy as np
import pandas as pd

train_data = np.array(DataFrame数据) #先将数据框转换为数组
train_data_list = train_data.tolist()  #其次转换为列表
print(np.array(train_data_list))  #以数组形式打出来方便看

0x02 List → DataFrame

列表转数据框根据需要有3中转换方式

方式1：传入单个数组

from pandas.core.frame import DataFrame
data_set = ['A','B','C']
data = DataFrame(data_set)  
#data = pd.DataFrame(data_set)
print(data)

执行结果：

0
0 A
1 B
2 C

方式2：传入字典

data_set = [{ 'IDCARD' :['ID1','ID1','ID1','ID2','ID2'],'TIME':['2019/11/21','2019/11/29','2019/11/30','2019/1/28','2019/2/13']}]
data = DataFrame(data_set)  
#data = pd.DataFrame(data_set)
print(data)

执行结果：

                  IDCARD                                               TIME

0 [ID1, ID1, ID1, ID2, ID2] [2019/11/21, 2019/11/29, 2019/11/30, 2019/1/28...

方式3：传入嵌套列表

data_set = [['ID1','ID1','ID1','ID2','ID2'],['2019/11/21','2019/11/29','2019/11/30','2019/1/28','2019/2/13']]
data = DataFrame(data_set)  
#data = pd.DataFrame(data_set)
print(data)

执行结果：

        0           1           2          3          4

0 ID1 ID1 ID1 ID2 ID2
1 2019/11/21 2019/11/29 2019/11/30 2019/1/28 2019/2/13