大家做在数据处理的时候,肯定难免会与文件交互,那么对于指定的文件类型,我们如何操作呢?
1.python读写csv文件
import csv
#python2可以用file替代open
with open('test.csv','w') as csvFile:
writer = csv.writer(csvFile)
#先写columns_name
writer.writerow(["index","a_name","b_name"])
#写入多行用writerows
writer.writerows([[1,2,3],[0,1,2],[4,5,6]])
#用reder读取csv文件
with open('test.csv','r') as csvFile:
reader = csv.reader(csvFile)
for line in reader:
print line
一定要注意,csv文件在写入时,字段和字段之间是用逗号“”,“”分割的,如果稍微不注意,就会出现串行的情况。
2.python读写excel文件
import xlwt
#创建工作簿
f = xlwt.Workbook()
'''
创建第一个sheet:
sheet1
'''
#创建sheet
sheet1 = f.add_sheet(u'sheet1',cell_overwrite_ok=True)
row0 = [u'编号',u'单位',u'网站备案号',u'域名',u'外链名称',u'不良网站名',u'不良url地址',u'地市',u'ip地址',u'截图',u'备注',u'审核时间',u'父链接',u'源码超链接']
#生成第一行
for i in range(0,len(row0)):
sheet1.write(0,i,row0[i]) # 顺序为x行x列写入第x个元素
f.save('newfile.xls')
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# 读取excel数据
import xlrd
data = xlrd.open_workbook('test.xls') # 打开xls文件
table = data.sheets()[0] # 打开第一张表
nrows = table.nrows # 获取表的行数
for i in range(nrows): # 循环逐行打印
if i == 0: # 跳过第一行
continue
print table.row_values(i)
3.python读写txt文件
#第一种
with open("test.txt","r","utf-8") as f:
# 为a+模式时,因为为追加模式,指针已经移到文尾,读出来的是一个空字符串。
ftext = f.read() # 一次性读全部成一个字符串
ftextlist = f.readlines() # 也是一次性读全部,但每一行作为一个子句存入一个列表
#第二种
f = open('test.txt', 'r', 'utf-8')
ftext = f.read() # 一次性读取完成
ftextlist = f.readlines() # 同上
fline = f.readline() # 只读取1行
f.close() # 关闭文件
# 写入txt
f = open('test.txt', 'ab', 'utf-8') # 在文件存在内容的时候追加
f = open('test.txt', 'wb', 'utf-8') # 文件存在时直接覆盖,没有就创建文件写入
f.write('aa') # 文件写入内容
f.close()
# 第二种
with open('test.txt', 'a') as f: # 道理同上
f.write('aa')
with open('test.txt', 'w') as f: # 道理同上
f.write('aa')
接下来说说with open 和 open的区别:
####区别:
用open去打开一个文件进行读取,假设文件不存在的话,比如打开文件 aaaa
如果文件不存在,open()函数就会抛出一个IOError的错误,并且给出错误码和详细的信息告诉你文件不存在!
如果文件打开成功,接下来,调用read()方法可以一次读取文件的全部内容,Python把内容读到内存,用一个str对象表示
>>> f.read()
'Hello, world!'
最后一步是调用close()方法关闭文件。文件使用完毕后必须关闭,因为文件对象会占用操作系统的资源,并且操作系统同一时间能打开的文件数量也是有限的:
f.close()
由于文件读写时都有可能产生IOError,一旦出错,后面的f.close()就不会调用。所以,为了保证无论是否出错都能正确地关闭文件,我们可以使用try … finally来实现:
try:
f = open('/path/to/file', 'r')
print(f.read())
finally:
if f:
f.close()
但是每次都这么写实在太繁琐,所以,Python引入了with语句来自动帮我们调用close()方法:
with open('/path/to/file', 'r') as f:
print(f.read())
这和前面的try … finally是一样的,但是代码更佳简洁,并且不必调用f.close()方法。
总结:以后读写文件都使用with open语句,不要再像以前那样用f = open()这种语句了!
作者:可爱的小虫虫