–copy大佬的,手打只是为了记牢、、菜鸡落泪
原链接
有看到更高级的,感觉学不会,摆烂,够用就行
1.利用numpy读取
import numpy as np
data=np.loadtxt(open("path.csv","rb"),delimiter=",",skiprows=n,usecols=[2,3])
delimiter 为分隔符,skiprows 是跳过前 n 行,usecols 是使用的列数
2.利用pandas读取
import pandas as pd
data=pd.read_csv(r'path.csv',seq=',',header='infer',usecols=[5])
array=data.value[:,:]#读取数据需要将data转成数组
seq 为分隔符 ,header是列名,是每一列的名字,如果header=1,将会以第二行作为列名,读取第二行以下的数据。
3.利用Python I/O读取文件
csv.reader
import csv
filename='path.csv'
data=[]
with open(filename) as csvfile:
csv_reader=csv.reader(csvfile) #使用csv.reader读取csvfile中的文件
header=next(csv_reader) #读取第一行每一列的标题
for row in csvreader: # 将csv 文件中的数据保存到data中
data.append(row[5]) # 选择某一列加入到data数组中
print(data)
csv.DictReader
import csv
with open(filename) as csvfile:
reader = csv.DictReader(csvfile)
column = [row['weight'] for row in reader] # weight 同列的数据
print(column)