【Python】数据框DataFrame和列表List相互转换-Toy模板网

这篇具有很好参考价值的文章主要介绍了【Python】数据框DataFrame和列表List相互转换。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

0x00 前言

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

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

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))  #以数组形式打出来方便看

【Python】数据框DataFrame和列表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)