AI算法工程师 | 02人工智能基础-Python基础（四）os模块

文章目录

• Python基础语法 之 文件和流
• • os 模块
  • • 调用 *** 作系统命令
    • *** 作文件和目录
    • 案例-文件重命名
  • 打开读取文件
  • • 打开文件
    • 读取文件
    • 补充-中文编码问题

Python基础语法 之 文件和流 os 模块

os 模块以及子模块 path 中包含了大量 *** 作文件和目录的函数。

调用 *** 作系统命令

os.system 可以直接调用系统的命令

示例：

• os.system 调用 windows 系统的记事本程序

import os
os.system("notepad.exe")  # 相当于在cmd中进行 notepad.exe  *** 作

• os.system 调用 windows 系统中 ping 命令

import os
print (os.system('ping www.baidu.com'))

*** 作文件和目录

在 os 模块中提供了一些 *** 作目录和文件的函数。部分函数的功能如下：

函数	功能	详情
getcwd 函数	获取工作目录	os.getcwd()：获取当前的工作目录，并输出该目录
mkdir 函数	创建目录	os.mkdir(目录名,目录的权限)：若指定的目录名存在，则抛出 OSError 异常
listdir 函数	列出文件	os.listdir(目录路径)：列出目录中的文件
rmdir 函数	删除目录	os.rmdir(dirname)：删除 dirname 参数指定的目录，若目录不为空，则抛出 OSError 异常
rename 函数	重命名	os.rename(src,dst)：将 src 参数指定的文件(目录)名改成 dst 参数指定的文件名

示例：

import os
d = 'D:/diu_test'  # 指明当前的文件夹
for file_name in os.listdir(d):   # os.listdir 列出文件夹下的文件
    print(file_name)    # 打印出列表中的元素

part-a.txt
part-b.txt
part-c.txt

知识补充：字符串中的方法

• replace 方法：替换某些字符（实际：创建了新的字符串对象，而不是修改以前的字符串）

a='Hello Python'
b=a.replace('o','*') # 将 Hello Python 字符串中的 o 替换为* 默认全部替换
c=a.replace('o','*',1) # 将 Hello Python 字符串中的 o 替换为* 替换 1 次
print(a)
print(b)
print(c)

Hello Python
Hell* Pyth*n
Hell* Python

• split 方法：通过分隔符对字符串进行切片，返回分隔后的字符串列表（分隔符：默认为所有空格字符 ）

a='Come rain or shine'
b=a.split(" ",2) # 按 " " 分隔，且仅分隔 2 个字符串
print(b)
c=a.split()  # 默认以所有空格字符进行分隔
print(c)

['Come', 'rain', 'or shine']
['Come', 'rain', 'or', 'shine']

知识补充：shutil 模块 之 move 方法

os 模块提供了对目录或文件的一般 *** 作。shutil 模块作为补充，提供了移动、复制、压缩、解压等 *** 作，这些 os 模块都没有提供。

shutil.move() 方法：可实现文件或目录的移动

语法格式：

shutil.move(src,des) 

move方法不同作用对应的参数含义：

作用	src 含义	des 含义
① 移动目录(文件夹)	是一个目录	a. 若 des 是一个存在的目录 =》会把 src 目录移动到 des 下面；
①		b. 若 des 是一个不存在的目录 =》会先创建该目录，再把 src 目录下面的所有文件移动到创建的 des 目录中
② 移动文件	是一个文件路径	a. des 可以是一个存在的文件夹路径 =》 将 src 原文件移动到 des 目录下；
②		b. 也可以是一个存在的文件夹路径加文件名 =》会将 src 原文件移动到 des 文件夹路径下，并重命名为 des 给的文件名

示例分析：

参考1：Python shutil.move() 用法及代码示例
参考2：shutil.move 移动文件或目录

案例-文件重命名

案例：通过Python，对电脑中的文件重命名

代码：

import os
import shutil

d = 'D:/diu_test'  # 指明文件夹（可用绝对路径，也可使用相对路径）
for file_name in os.listdir(d):   # os.listdir 列出文件夹下的文件
    print(file_name)    # 打印出列表中的元素
    new_file_name = file_name.replace('part', 'part-r-0000') # 将'part'部分 替换成  'part-r-0000' 后返回新的字符串
    print(new_file_name)
    shutil.move(d+'/'+file_name, d+'/'+new_file_name)  # 重命名(move：移动文件。由于在同一个路径下，所以实现了重命名的效果)
print('Done')

part-a.txt
part-r-0000-a.txt
part-b.txt
part-r-0000-b.txt
part-c.txt
part-r-0000-c.txt
Done

打开读取文件 打开文件

open() 方法：打开一个文件，并返回文件对象

文件模式	含义	具体
w	以写的方式打开文件	write
a	以追加的方式打开文件	append
r	以读的方式打开文件	read

示例：写入文件

myfile = '\n做你想做的，趁阳光正好\n'
f = open('./test/myfile.txt', 'a') # 以追加的方式打开文件（若无文件，会新建该文件）
f.write(myfile) # 将内容写入文件中
f.close()

读取文件

read() 方法：可一次读取文件的全部内容（Python 把内容读到内存）

示例：

• read() 方法读取文件（并使用 close() 关闭文件 ）

f = open('./test/myfile.txt', 'r') # 以读的方式打开文件
data = f.read()  # 读取文件
print(data)
f.close()

做你想做的，趁阳光正好

做你想做的，趁阳光正好

• 通过 with 语句实现文件的读取（优点：无需使用 close() 关闭文件，系统会自动关闭 ）

# with 方式：会反复调用 read(size)方法（size 是一个整数，表示从文件指针指定的位置）。
# 可避免因一次性读取文件的全部内容，而导致内存不足的问题
开始读取 size 个字节。
with open('./test/myfile.txt', 'r') as f:
    data = f.read()
    print(data)

做你想做的，趁阳光正好

做你想做的，趁阳光正好

readline 与 readlines 方法：读行

示例：

• readline() 方法：每次读出一行内容（∴ 读取时占用的内存小，比较适合大文件）

# readline 方法：用于从文件指针当前位置读取一整行文本，
# 遇到行结束符停止读取文本，但读取的内容包括了行结束符。

f = open('./test/myfile.txt', 'r')
line = f.readline() # 先读一行，如果能读到内容，则循环让其读下一行
while line:      # 通过循环读取每一行
    print(line)
    line = f.readline()
f.close()

做你想做的，趁阳光正好



做你想做的，趁阳光正好

• readlines() 方法：读取整个文件所有行，保存在一个列表变量中（每行对应列表的一个元素，但是读取大文件会比较占内存）

# readlines 方法：从文件指针当前的位置读取后面的所有数据，
# 并将这些数据按行结束符分隔后，放到列表中返回。

f = open('./test/myfile.txt', 'r')
lines = f.readlines()
for line in lines:
    print(line)
f.close()

做你想做的，趁阳光正好



做你想做的，趁阳光正好

补充-中文编码问题

指定文件编码解决中文乱码问题：

myfile = '\n做你想做的，趁阳光正好\n'
f = open('./test/myfile.txt', 'w', encoding='utf-8') # 可添加编码格式
f.write(myfile)
f.close()

f = open('./test/myfile.txt', 'r', encoding='utf-8') # 与写入时统一编码
line = f.readline()
while line:
    print(line)
    line = f.readline()
f.close()

做你想做的，趁阳光正好

—— 说明：本文内容基于python3.0