首先使用np.array()函数把DataFrame转化为np.ndarray(),再利用tolist()函数把np.ndarray()转为list,示例代码如下:
# -*- coding:utf-8-*-
import numpy as np
import pandas as pd
data_x = pd.read_csv("E:/Tianchi/result/features.csv",usecols=[2,3,4])#pd.dataframe
data_y = pd.read_csv("E:/Tianchi/result/features.csv",usecols=[5])
笔者在学习pandas,在学习过程中总结了一下创建dataframe的方法,通过查阅资料总结遗下几种方法,如果你有其他的方法欢迎留言补充。
练习代码 请点击此处下载
学习环境:
第一种: 用Python中的字典生成
第二种: 利用指定的列内容、索引以及数据
第三种:通过读取文件,可以是json,csv,excel等等。
本文例子就用excel, 上篇博客笔者已经用csv举例了。这里要注意,如果用excel请先安装xlrd这个包
这个文件笔者放在代码同目录
第四种:用numpy中的array生成
第五种: 用numpy中的array,但是行和列名都是从numpy数据中来的
import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})
array = df.values
转换后的数组类型为numpy数组。
【调试问题】RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn
79566