分割csv文件代码,Python代码库,德仔网

代码语言

代码分类

【Python】分割csv文件

作者:taota / 发布于2015/7/9/ 463

cmd运行 python （文件名）（分割数）（可选分割符转转成csv文件并分割）


	# coding:utf-8
	# python 2.7
	import csv
	import math
	import sys
	import os
	from sys import exit
	
	#平均分割CVS文件
	def deal_csv(head,data,n,filepath):
	 name = filepath.split('.')[0]
	 line_number = 1000000
	 if int(math.ceil(len(data)/float(n)))<=line_number:
	 line_number = int(math.ceil(len(data)/float(n)))
	 else:
	 n = int(math.ceil(len(data)/float(line_number)))
	 temp = []
	 for i in range(len(data)-1,-1,-1):
	 temp.append(data[i])
	 if i==(n-1)*line_number:
	 reader = csv.writer(open("%s_%d.csv"%(name,n),'wb'))
	 reader.writerow(head)
	 for line in temp:
	 reader.writerow(line)
	 n-=1
	 temp=[]
	 print 'SourceFile -> %s'%(name)
	 print 'FilePath -> %s'%(os.getcwd())
	 print 'Status -> Success'
	#读取文件内容并按行转化成数组
	def read_csv(filepath,sign):
	 reader = csv.reader(file(filepath,'rb'))
	 data = []
	 if sign==',':
	 for line in reader:
	 data.append(line)
	 else:
	 for line in reader:
	 data.append(line[0].split(sign))
	 return data
	def read_file(filepath,sign):
	 file = open(filepath)
	 try:
	 data = []
	 reader = file.readlines()
	 for line in reader:
	 data.append(line.strip('\n').split(sign))
	 return data
	 finally:
	 file.close()
	
	if __name__=="__main__":
	 data = []
	 filepath=''
	 sign=''
	 if len(sys.argv)==3:
	 sign=','
	 elif len(sys.argv)==4:
	 sign=sys.argv[3]
	 else:
	 sys.exit()
	 #获取文件名
	 filepath=sys.argv[1]
	 #获取数据
	 data = read_file(filepath,sign)
	 #获取分割数
	 n=int(sys.argv[2])
	 deal_csv(data[0],data[1:],n,filepath)

评论列表

本站所提供的代码，版权归原作者所有，若有侵犯作者版权，请与我们联系，我们将立即删除或修改。谢谢!
本站所有代码发布及提供者。

试试其它关键字

　分割　　 csv文件　

同语言下

可能有用的

taota贡献的其它代码(1)

.分割csv文件