pandas DataFrame数据转为list

首先使用np.array()函数把DataFrame转化为np.ndarray()，再利用tolist()函数把np.ndarray()转为list，示例代码如下：

# -*- coding:utf-8-*-import numpy as npimport pandas as pddata_x = pd.read_csv("E:/Tianchi/result/features.csv",usecols=[2,3,4])#pd.dataframedata_y =  pd.read_csv("E:/Tianchi/result/features.csv",usecols=[5])train_data = np.array(data_x)#np.ndarray()train_x_list=train_data.tolist()#listprint(train_x_list)print(type(train_x_list))

如果你觉得本文有用，可以给作者打赏一下。