@TOCpython 读写csv文件转二维数组(不用numpy)
工作需要处理CSV中的数据,需将数据转为二维数组进行处理,无奈Python版本没有numpy这个包,研究一天,试出两种方法:
import csv
import sys
#第一种通用方法:#
file = open('input','r')
line = file.readline()
data = []
while line:
list_arr = []
line = line.strip()#去除空格换行等字符#
line = line.split(',')#按照逗号分组#
arr = list(map(int,line))#将stringList转为一维数组#
data.append(arr)
line = file.readline()
file.close
rowNum = len(data)
columnNum = len(data[0])
for i in range(columnNum):
#l处理数据的逻辑#
sys.argv[1]#用来接收控制台输入数据#
with open('path','w',newline='') as f:
writer = csv.writer(f)
writer.writerows(data)
f.close
#第二种只针对csv:#
with open(path,'r') as csvfile:
reader = csv.reader(csvfile)
data = []
for line in reader:
data.append(line)
由于没有numpy这个架包,难点主要在于数据转换,如何转为二维数组,第二种更简单一点