Python中os和csv模块的使用

目录

os模块的使用

open("test.txt","mode")

读取文件中的内容 

 f.read()

 f.readline(size)  

 f.readlines(size) 

写入数据到文件中

追加数据

覆盖原数据

读写文件的IO指针的一些问题 

os模块中一些基本的判断文件的函数

OS模块执行系统命令 

CSV模块的使用

读文件

reader(csvfile, dialect='excel', **fmtparams)

读取指定的列

DictReader函数是返回字典格式的数据

读取指定列的数据 

写文件

writer(csvfile, dialect='excel', **fmtparams)

追加一行数据进去

覆盖之前的数据


OS模块简单的来说它是一个Python的系统编程的操作模块,可以处理文件和目录

以下都是python3.7环境下

os模块的使用

open("test.txt","mode")

mode有以下几种:

  • r : 只读,如果无此文件的话就会报错
  • r+:   读写方式打开,既可读也可写
  • w: 写入,覆盖之前的内容,如果无此文件就新建,只能写不能读
  • a:  写入,在文件末尾追加新的内容,文件如果不存在,则新建,只能写不能读
  • +: 更新,须结合 r、w、a 参数使用,指针在最后面
  • b: 打开二进制的文件,可以与 r、w、a、+ 结合使用
  • U:支持所有的换行符号。“ \r ”、“ \n ", " \r\n "

注:r 、w 、a 不能结合使用。不写mode参数默认是只读

两种打开文件的方式

f=open("test.txt","r")
或
with open("test.txt","r") as f:

 如果是读取其他路径下的文件的话

open("c://users//desktop//test.txt")
open("c:\\users\\desktop\\test.txt")
open("c:\/users\/desktop\/test.txt")

现在有一个 test.txt 文件,里面有以下内容

hello,word!
iloveyou
xiaoxie

读取文件中的内容 

不管用什么方法读取文件中的内容,只要不是文件的最后一行,其他行的末尾都会有一个 \n 。上面文件的内容,如果读取的话,在读取的字符串中实际上是这样的。所以我们在读取文件内容的时候,都会对文件去除\n ,使用字符串的 strip() 方法

hello,word!\n
iloveyou\n
xiaoxie

 f.read()

函数,读取文件中的所有内容,返回一个 str 字符串

import os 
try:
    with open("test.txt",'r',encoding="utf-8") as f:
        a=f.read()
        print(a)
        print(type(a))
        print(len(a))
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()
##########################################################
hello,word!
iloveyou
xiaoxie
<class 'str'>
28

明明是26字节,为什么打印出来是28字节呢?这是因为第一行和第二行的 \n 也被算成了一个字符

 f.readline(size)  

函数,读取文件的第一行内容,返回一个字符串,参数size代表读取的字符串长度,不填默认读取所有第一行所有内容,包括 \n

import os 
try:
    with open("test.txt",'r',encoding="utf-8") as f:
        a=f.readline()             #读取文件的第一行
        print(a)
        print(type(a))
        print(len(a))
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()
####################
hello,word!

<class 'str'>
12

这里为什么第二行是空的呢,字符串长度是12呢?这是因为第一行末尾还有一个 \n , 这是换行符,所以第二行是空的,也所以字符串长度读成了12,要想改成,我们可以修改下面几行,这样打印出来的就正常了

print(a.strip('\n'))
print(type(a))
print(len(a.strip('\n')))
########
hello,word!
<class 'str'>
11

 f.readlines(size) 

函数读取文件中的内容,参数size代表读取的行数,不填默认读取所有行,返回一个 list 列表,每一行作为列表的一个值

这里要注意的是,如果要打印出每行的内容的话,要用 str.strip() 方法把末尾的 \n 给去掉

import os 
try:
    with open("test.txt",'r',encoding="utf-8") as f:
        lines=f.readlines()
        print(lines,type(lines))
        for line in lines:
            print(line.strip('\n'),end='---')
            print(len(line.strip('\n')))
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()
######################
['hello,word!\n', 'iloveyou\n', 'xiaoxie'] <class 'list'>
hello,word!---11
iloveyou---8
xiaoxie---7

写入数据到文件中

  • f.write(str) 的参数是一个字符串,就是你要写入的内容
  • f.writelines(sequence)的参数是序列,比如列表,它会迭代帮你写入文件。但是序列中的元素也必须是str字符串

追加数据

默认我们写入的数据是在原数据的末尾的,也就是不是新的一行。如果我们想一行一行写入,我们可以在写入每一行之前,先写入 \n

import os 
try:
    with open("test.txt",'a',encoding='utf-8') as f:
        f.write("nihao"+"\n")
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()  

覆盖原数据

import os 
try:
    with open("test.txt",'w',encoding='utf-8') as f:
        f.write("nihao"+"\n")
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()

还有一种利用print写入数据,知道就好,不常用

import os
f=open("test.txt","w",encoding='utf-8')
print("hello,word!",file=f)   #将hello,word!追加写入test.txt文件中
f.close()

读写文件的IO指针的一些问题 

比如我们有一个文件 test.txt 里面的内容是: hello,word!

import os 
with open("test.txt","r",encoding="utf-8") as f:
    print("*"*50)
    print(f.read())
    print("*"*50)
    print(f.read())
    print("*"*50)
    f.close()

**************************************************
hello,word!
**************************************************

**************************************************

我们可以看到,第二次打印什么东西也没打印出来。那是因为我们第一次读取文件的时候,我们的指针已经到文件的末尾了,当我们再次读取文件的时候,自然就没有东西了。除非我们关闭文件流,再次打开读取。 

os模块中一些基本的判断文件的函数

os.getcwd()          返回当前工作路径
os.chdir("e://")     修改工作路径
   
open("test.txt","w")       创建文件
os.remove("test.txt")      删除文件


os.mkdir("e://test")           在当前目录下创建目录
os.rmdir("e://test")           删除给定路径的目录(该目录必须是空目录,不然报错)
os.removedirs("e://test")      删除给定路径的目录(该目录必须是空目录,不然报错)


os.listdir("e://")    返回给定路径下的所有文件和目录,返回的是列表类型的数据

os.path.abspath("123.txt")  获取当前目录下某文件的绝对路径,如果该文件不存在,也返回路径
 
os.path.exists("123")   判断该目录下文件或者目录是否存在,如果存在返回true,否则返回false

如果存在,则再判断是目录还是文件,如果存在返回true,否则返回false
os.path.isdir("123")     判断是否是目录
os.path.isfile("123")   判断是否是文件
 
os.path.isabs("c://users//")   判断给定的路径是绝对路径还是相对路径,绝对路径是话返回True,相对路径返回False。他不会去判断你给定的路径是否真实存在

os.path.split("c://users//test.txt")  将给定路径的路径和文件名分隔开,返回元组型数据  ('c://users', 'test.txt')

os.path.split("c://users//test.txt")[0]  返回给定路径的路径名   c://users
os.path.dirname("c://users/test.txt")    返回给定路径的路径名   c://users

os.path.split("c://users//test.txt")[1]  返回给定路径的文件名   test.txt
os.path.basename("c://users//test.txt")  返回给定路径的文件名   test.txt

path=os.path.join("c:// ",123.txt)  将路径和文件名加在一起

获取当前目录下指定后缀的文件 

import os
cur_path=os.getcwd()
filenames=os.listdir(cur_path)
for filename in filenames:
	if os.path.splitext(filename)[1] == ".txt":
		print(filename)

 实例:递归打印出给定路径下的所有文件

#递归打印出给定路径内的所有文件
import os 
def Test(path):
    lines=os.listdir(path)
    for line in lines:
        pathname=os.path.join(path,line)
        if os.path.isdir(pathname):              #如果是目录的话
            print("这是目录:",pathname)
            Test(pathname)                      #再递归调用该函数
        else:
            print(pathname)                     #如果是文件
Test("g://xie")

OS模块执行系统命令 

os模块执行命令有以下几个:

  • os.system(command):执行成功返回0,执行失败返回1
  • os.popen(command):这种调用方式是通过管道的方式来实现,函数返回是 file read 的对象,对其进行读取read、readlines等操作可以看到执行的输出,如果命令执行失败,就读取不到内容。返回的是一个对象的类型
import os 
result=os.system("whoami")
print("*"*30)
print(result)

import os 
result=os.popen("ipconfig")
print(result)
print("*"*30)
result2=os.popen("ipconfig").readlines()
print(result2)
print("*"*30)
for line in result2:
    print(line.strip("\n"))

CSV模块的使用

CSV (Comma Separated Values) 即逗号分隔值(也称字符分隔值,因为分隔符可以不是逗号),是一种常用的文本格式,用以存储表格数据,包括数字或者字符。很多程序在处理数据时都会碰到csv这种格式的文件,它的使用是比较广泛的,python内置了处理csv格式数据的csv模块。

读文件

比如我们现在有一个 test.csv的文件,文件内容如下

我们使用reader函数来读取 test.csv 文件中的内容

reader(csvfile, dialect='excel', **fmtparams)

  • csvfile 参数,必须是支持迭代(Iterator)的对象,可以是文件(file)对象或者列表(list)对象,如果是文件对象,打开时需要加"b"标志参数。
  • dialect参数,编码风格,不写则默认为excel的风格,也就是用逗号(,)分隔,dialect方式也支持自定义,通过调用register_dialect方法来注册
  • fmtparams参数,格式化参数,用来覆盖之前dialect对象指定的编码风格。
import csv
try:
    with open("test.csv","r",encoding='utf-8') as f:
        lines=csv.reader(f)        #返回的是 _csv.reader 对象
        print(lines,type(lines))
        for line in lines: 
            print(line,type(line))   #返回的每一行是列表格式
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()
#########################
<_csv.reader object at 0x00000141BEC9E938> <class '_csv.reader'>
['id', 'name', 'age'] <class 'list'>
['1', 'xie', '20'] <class 'list'>
['2', 'xiao', '30'] <class 'list'>
['3', 'wang', '40'] <class 'list'>
['4', 'li', '50'] <class 'list'>

读取指定的列

import csv
try:
    with open("test.csv","r",encoding='utf-8') as f:
        lines=csv.reader(f)
        for line in lines:
            print(line[1],type(line[1]))     ##返回的是字符串
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()
##########################
name <class 'str'>
xie <class 'str'>
xiao <class 'str'>
wang <class 'str'>
li <class 'str'>

DictReader函数是返回字典格式的数据

import csv
try:
    with open("test.csv","r") as f:
        lines=csv.DictReader(f)        #返回的是 _csv.reader 对象
        print(lines,type(lines))
        for line in lines: 
            print(line,type(line))   #返回的每一行是列表格式
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()
##########################
<csv.DictReader object at 0x000001E63C477668> <class 'csv.DictReader'>
OrderedDict([('id', '1'), ('name', 'xie'), ('age', '20')]) <class 'collections.OrderedDict'>
OrderedDict([('id', '2'), ('name', 'xiao'), ('age', '30')]) <class 'collections.OrderedDict'>
OrderedDict([('id', '3'), ('name', 'wang'), ('age', '40')]) <class 'collections.OrderedDict'>
OrderedDict([('id', '4'), ('name', 'li'), ('age', '50')]) <class 'collections.OrderedDict'>

读取指定列的数据 

import csv
try:
    with open("test.csv","r") as f:
        lines=csv.DictReader(f)        #返回的是 _csv.reader 对象
        col=[line['name'] for line in lines]
        print(col,type(col))
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()
##################################
['xie', 'xiao', 'wang', 'li'] <class 'list'>

写文件

writer(csvfile, dialect='excel', **fmtparams)

  • csvfile 参数,必须是支持迭代(Iterator)的对象,可以是文件(file)对象或者列表(list)对象
  • dialect参数,编码风格,默认为excel的风格,也就是用逗号(,)分隔,dialect方式也支持自定义,通过调用register_dialect方法来注册
  • fmtparams参数,格式化参数,用来覆盖之前dialect对象指定的编码风格。

写入数据时传入的数据,可以是元组,也可以是列表,他会自动把元组或列表中的数据写进不同列中。

追加一行数据进去

import csv
row=['5','zhang','60']
try:
    with open("test.csv","a",newline='',encoding='utf-8') as f:
       csv_writer=csv.writer(f)
       csv_writer.writerow(row)
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()

注:python3 和 python2 这个模块的使用有点不同。python3写入为了防止空一行,会在加一个 newline 参数。python2的话,则是写入方式加一个b

python3:with open("test.csv","a",newline='') as f:
python2:with open("test.csv","ab") as f:

覆盖之前的数据

import csv
try:
    with open("test.csv","w",newline='',encoding='utf-8') as f:
       csv_writer=csv.writer(f)
       csv_writer.writerow(["id","name","age"])
       csv_writer.writerow(["1","xie","20"])
       csv_writer.writerow(["2","wang","30"])
except Exception as e:
    print("异常对象的类型是:%s"%type(e))
    print("异常对象的内容是:%s"%e)
finally:
    f.close()

 

已标记关键词 清除标记
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 19.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值