首先使用np.array()函数把DataFrame转化为np.ndarray(),再利用tolist()函数把np.ndarray()转为list,示例代码如下:
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])
train_data = np.array(data_x)#np.ndarray()
train_x_list=train_data.tolist()#list
print(train_x_list)
print(type(train_x_list))
---------------------
作者:nudt_qxx
来源:CSDN
原文:https://blog.csdn.net/xiangxianghehe/article/details/72615711
版权声明:本文为博主原创文章,转载请附上博文链接!
# ---------------------综合小例子----------------------------
col_names = food_info.columns.tolist() #把所有的行转化成list
print(col_names)
gram_columns = []
for c in col_names: #遍历col_names,找出所有以(g)结尾的位置
if c.endswith("(g)"):
gram_columns.append(c)
print(gram_columns)
gram_df = food_info[gram_columns] #把所有以(g)结尾的列存放到gram_df
print(gram_df.head(3))
---------------------
作者:数据阿伯
来源:CSDN
原文:https://blog.csdn.net/qq_32005671/article/details/61199628
版权声明:本文为博主原创文章,转载请附上博文链接!