一、pandas读入csv文件
1、pd.read_csv()读入csv文件。
import numpy as np
data = pd.read_csv('./input/covid.train.csv')
x = data[data.columns[1:94]]
y = data[data.columns[94]]
2、columns用法
使用pandas处理列表的时候,调用上面的表头信息。
(1)建立一个excel形式文件
df = pd.DataFrame([[1, 2, 3]], columns=list("ABC"))
输出:
A B C
0 1 2 3
(2)获得表头信息:
col = df.columns # 获取到的col是<class 'pandas.core.indexes.base.Index'>
输出
Index(['A', 'B', 'C'], dtype='object')
(3)获得列表形式表头
(3)col = df.columns.values # 返回结果是数组
输出
['A' 'B' 'C']