Python文件操作大全

1 文件操作

1.1 文件打开与关闭

1.1.1 打开文件

在Python中,你可以使用 open() 函数来打开文件。以下是一个简单的例子:

python 复制代码
# 打开文件(默认为只读模式)
file_path = 'example.txt'
with open(file_path, 'r') as file:
    # 执行文件操作,例如读取文件内容
    file_content = file.read()
    print(file_content)

# 文件在with块结束后会自动关闭,无需显式关闭文件

在上述示例中:

'example.txt' 是文件的路径和名称,你可以根据实际情况修改为你想要打开的文件。

'r' 表示只读模式。如果你想要写入文件,可以使用 'w' 模式,如果想要追加内容,可以使用 'a' 模式等。

with open(...) as file : 是使用上下文管理器的方式,确保文件在使用后被正确关闭,即使在处理文件时发生异常也能保证关闭。

1.1.2 关闭文件

在 Python 中关闭文件有两种主要的方法:

1. 使用 with 语句:

with 语句是一种上下文管理器,当它的代码块执行完毕时,会自动关闭文件。这是推荐的方式,因为它确保文件在使用完毕后被正确关闭,即使发生异常也能保证关闭。

python 复制代码
   file_path = 'example.txt'
   with open(file_path, 'r') as file:
       # 执行文件操作,例如读取文件内容
       file_content = file.read()
       print(file_content)
   # 文件在这里已经被自动关闭

2. 使用 close() 方法:

你可以显式调用文件对象的 close() 方法来关闭文件。这种方法适用于一些特殊情况,但相对来说不如 with 语句简洁和安全。

python 复制代码
 file_path = 'example.txt'
   file = open(file_path, 'r')
   try:
       # 执行文件操作,例如读取文件内容
       file_content = file.read()
       print(file_content)
   finally:
       file.close()

在使用 with 语句时,不需要显式调用 close() 方法。如果你在代码中打开了文件而没有使用 with,请确保在适当的地方调用 close() 以关闭文件,以避免资源泄漏。

1.2 访问模式及说明

2 文件读写

2.1 写数据(write)

写入数据通常涉及将信息保存到文件、数据库或其他持久性存储介质中。以下是一些常见的数据写入场景的示例:

1. 写入文本文件

使用内置的 open 函数来打开文件并写入内容。确保使用适当的模式(例如,'w' 表示写入)。

python 复制代码
file_path = 'example.txt'

# 写入文件
with open(file_path, 'w') as file:
    file.write("Hello, this is some data.")

2. 写入CSV文件

使用 csv 模块来写入CSV格式的文件。

python 复制代码
import csv

csv_file_path = 'example.csv'

data = [['Name', 'Age', 'Occupation'],
        ['John Doe', 30, 'Engineer'],
        ['Jane Smith', 25, 'Designer']]

with open(csv_file_path, 'w', newline='') as csvfile:
    csv_writer = csv.writer(csvfile)
    csv_writer.writerows(data)

3. 写入JSON文件

使用内置的 json 模块来写入JSON格式的文件。

python 复制代码
import json

json_file_path = 'example.json'

data = {"name": "John Doe", "age": 30, "occupation": "Engineer"}

with open(json_file_path, 'w') as jsonfile:
    json.dump(data, jsonfile)

4. 写入数据库

使用数据库连接库(如 sqlite3、mysql-connector-python 等)与相应的数据库进行交互。

python 复制代码
import sqlite3

# 连接到SQLite数据库(假设有一个名为 example.db 的数据库)
conn = sqlite3.connect('example.db')

# 创建一个游标对象
cursor = conn.cursor()

# 执行SQL插入语句
cursor.execute("INSERT INTO users (name, age, occupation) VALUES (?, ?, ?)", ('John Doe', 30, 'Engineer'))

# 提交更改
conn.commit()

# 关闭连接
conn.close()

2.2 读数据(read)

读取数据通常涉及从文件、数据库或其他存储介质中检索信息。以下是一些读取数据的常见示例:

1. 读取文本文件

使用内置的 open 函数来打开文件并读取内容。

python 复制代码
file_path = 'example.txt'

# 读取文件
with open(file_path, 'r') as file:
    data = file.read()
    print(data)

2. 读取CSV文件

使用 csv 模块来读取CSV格式的文件。

python 复制代码
import csv

csv_file_path = 'example.csv'

# 读取CSV文件
with open(csv_file_path, 'r') as csvfile:
    csv_reader = csv.reader(csvfile)
    for row in csv_reader:
        print(row)

3. 读取JSON文件

使用内置的 json 模块来读取JSON格式的文件。

python 复制代码
import json

json_file_path = 'example.json'

# 读取JSON文件
with open(json_file_path, 'r') as jsonfile:
    data = json.load(jsonfile)
    print(data)

4. 从数据库中读取数据

使用数据库连接库(如 sqlite3、mysql-connector-python 等)与相应的数据库进行交互。

python 复制代码
import sqlite3

# 连接到SQLite数据库(假设有一个名为 example.db 的数据库)
conn = sqlite3.connect('example.db')

# 创建一个游标对象
cursor = conn.cursor()

# 执行SQL查询语句
cursor.execute("SELECT * FROM users")

# 检索所有行
rows = cursor.fetchall()

# 打印每一行
for row in rows:
    print(row)

# 关闭连接
conn.close()

2.3 读数据(readlines)

readlines 是 Python 中用于读取文件的方法之一,它用于逐行读取文件内容,并将每一行作为字符串存储在一个列表中。下面是对 readlines 方法的详细解释:

使用 readlines 方法的基本语法

python 复制代码
with open('file.txt', 'r') as file:
    lines = file.readlines()

解释:

open('file.txt', 'r') : 打开文件 'file.txt' 以供读取。第一个参数是文件名,第二个参数是打开文件的模式。'r' 表示只读模式。

with ... as ... : 使用 with 语句可以确保在读取完成后自动关闭文件,不需要显式调用 file.close()。

lines = file.readlines() : readlines 方法用于读取文件的所有行,并将每一行作为一个字符串存储在列表 lines 中。

每个列表元素对应文件中的一行文本。你可以使用列表索引来访问特定行,例如 lines[0] 表示文件的第一行。

例子:假设 'file.txt' 包含以下内容:

python 复制代码
Hello, this is line 1.
This is line 2.
And this is line 3.

使用 readlines 后:

python 复制代码
with open('file.txt', 'r') as file:
    lines = file.readlines()

# lines 现在是一个包含每一行文本的列表
print(lines)
# 输出:
# ['Hello, this is line 1.\n', 'This is line 2.\n', 'And this is line 3.\n']

# 访问特定行
print(lines[0].strip())  # 输出:Hello, this is line 1.

注意事项:

每一行的末尾都包含换行符 \n,你可以使用 strip() 方法去除这些额外的空白字符。

readlines 方法适用于处理包含多行文本的文件,但对于大型文件,可能需要考虑逐行读取而不是将整个文件加载到内存中。这可以通过循环遍历文件对象来实现,而不是使用 readlines。

2.3 读数据(readline)

readline 是 Python 中用于读取文件的方法之一,它用于逐行读取文件内容,并返回文件中的一行作为字符串。以下是对 readline 方法的详细解释:

使用 readline 方法的基本语法

python 复制代码
with open('file.txt', 'r') as file:
    line = file.readline()

解释:

open('file.txt', 'r') : 打开文件 'file.txt' 以供读取。第一个参数是文件名,第二个参数是打开文件的模式。'r' 表示只读模式。

with ... as ... : 使用 with 语句可以确保在读取完成后自动关闭文件,不需要显式调用 file.close()。

line = file.readline() : readline 方法用于读取文件的一行,并将该行作为一个字符串存储在变量 line 中。

例子:假设 'file.txt' 包含以下内容:

python 复制代码
Hello, this is line 1.
This is line 2.
And this is line 3.

用 readline 后:

python 复制代码
with open('file.txt', 'r') as file:
    line1 = file.readline()
    line2 = file.readline()
    line3 = file.readline()

print(line1)  # 输出:Hello, this is line 1.
print(line2)  # 输出:This is line 2.
print(line3)  # 输出:And this is line 3.

注意事项:

每个 readline 调用都会读取文件的下一行。

返回的字符串包含行末尾的换行符 \n。如果不需要换行符,可以使用 strip() 方法去除它。

当文件读取完毕后,readline 将返回空字符串 '',因此可以在循环中使用 while line != '' 来逐行读取整个文件。

循环读取整个文件:

python 复制代码
with open('file.txt', 'r') as file:
    line = file.readline()
    while line != '':
        print(line.strip())  # 去除换行符
        line = file.readline()

这个循环将逐行读取整个文件,直到文件末尾。

2.4 readlines 和 readline的区别

readlinesreadline 是 Python 中用于读取文件的两种不同方法,它们之间有一些重要的区别:

1. readlines 方法:

  • 返回类型:readlines 方法返回一个包含文件所有行的列表,其中每个元素都是文件中的一行文本字符串。
  • 使用情况: 适用于处理包含多行文本的文件,可以一次性将整个文件加载到内存中。这种方法适用于文件较小,可以完全装入内存的情况。
  • 例子:
python 复制代码
  with open('file.txt', 'r') as file:
      lines = file.readlines()

2. readline 方法:

返回类型: readline 方法每次调用只返回文件中的一行作为字符串。如果再次调用,将返回下一行。当文件读取完毕后,返回空字符串 ''。

使用情况: 适用于逐行处理大型文件,可以有效地降低内存使用。因为它一次只读取一行,可以在循环中逐行处理文件,而不必将整个文件加载到内存中。

例子:

python 复制代码
  with open('file.txt', 'r') as file:
      line = file.readline()
      while line != '':
          print(line.strip())  # 去除换行符
          line = file.readline()

3. 区别总结:

readlines 一次性读取整个文件的所有行,并返回一个包含所有行的列表。

readline 逐行读取文件,每次调用返回文件中的一行,适用于处理大型文件,减少内存占用。

readlines 返回包含换行符的每一行,而 readline 返回单独的行,需要手动去除换行符。

选择使用哪个方法取决于文件的大小和处理需求。如果文件较小,可以完全装入内存,使用 readlines;如果文件较大,可以逐行处理,使用 readline。

3 文件的相关操作

3.1 文件重命名

Python 文件重命名是文件管理中的一个基本操作,可以通过 Python 的内置库来实现。以下是一个超详细的入门指南,介绍如何使用 Python 重命名文件:

1. 导入必要的库

首先,您需要导入 Python 的 os 库,它提供了许多与操作系统交互的函数。

python 复制代码
import os
2. 准备文件列表

要重命名文件,您需要先列出指定目录中的所有文件。可以使用 os.listdir() 函数来获取目录中的文件列表。

python 复制代码
# 列出指定目录中的所有文件和文件夹
files = os.listdir('path_to_directory')
3. 遍历文件列表
python 复制代码
for file in files:
    # 获取文件的完整路径
    full_path = os.path.join('path_to_directory', file)
    
    # 检查是否是文件
    if os.path.isfile(full_path):
        # 新的文件名
        new_filename = 'new_name'
        
        # 重命名操作
        os.rename(full_path, os.path.join('path_to_directory', new_filename))
        print(f'Renamed {file} to {new_filename}')
4. 异常处理

在重命名文件时,可能会出现各种异常,例如目标文件已存在、没有足够权限等。为了确保程序的健壮性,应该添加异常处理。

python 复制代码
try:
    for file in files:
        # ...(上面的代码)
except OSError as e:
    print(f'Error occurred: {e}')
5. 完整的脚本示例
python 复制代码
import os
# 指定要重命名文件的目录
directory = 'path_to_directory'
# 列出目录中的所有文件
files = os.listdir(directory)
# 遍历文件列表并进行重命名
for file in files:
    if os.path.isfile(os.path.join(directory, file)):
        # 设定新的文件名
        new_filename = 'new_name'
        
        # 重命名文件
        try:
            os.rename(
                os.path.join(directory, file),
                os.path.join(directory, new_filename)
            )
            print(f'Renamed {file} to {new_filename}')
        except OSError as e:
            print(f'Error renaming {file}: {e}')
6. 注意安全性和效率

在批量重命名文件时,应确保:

  • 不要同时进行多个重命名操作,以避免潜在的竞争条件。
  • 确保目标目录存在,避免在重命名时创建不存在的目录。
  • 考虑到操作系统对文件重命名的限制,例如在 Windows 中,文件名不能超过 255 个字符,而在 Unix/Linux 中则没有这个限制。
7. 高级用法

对于更复杂的重命名任务,您可以使用正则表达式或者其他文本处理方法来生成新的文件名。

python 复制代码
import os
import re
# 指定目录
directory = 'path_to_directory'
# 列出目录中的所有文件
files = os.listdir(directory)
# 遍历文件列表并进行重命名
for file in files:
    if os.path.isfile(os.path.join(directory, file)):
        # 使用正则表达式匹配文件名模式,并替换为新的模式
        new_filename = re.sub(r'\d+', 'new_prefix', file)
        
        # 重命名文件
        try:
            os.rename(
                os.path.join(directory, file),
                os.path.join(directory, new_filename)
            )
            print(f'Renamed {file} to {new_filename}')
        except OSError as e:
            print(f'Error renaming {file}: {e}')

这个脚本会将指定目录中所有以数字开头的文件重命名为新的前缀。

3.2 删除文件

在Python中,删除文件是一个相对简单的操作。我们可以使用os库中的os.remove()函数来实现。以下是一个超详细的入门指南,介绍如何使用Python删除文件:

1. 导入必要的库

首先,您需要导入Python的 os 库,它提供了许多与操作系统交互的函数。

python 复制代码
import os
2. 准备文件路径

要删除文件,您需要知道要删除的文件的路径。

python 复制代码
file_path = 'path_to_file'
3. 检查文件是否存在

在删除文件之前,最好检查该文件是否存在,以避免错误。

python 复制代码
if os.path.isfile(file_path):
    print(f'File {file_path} exists, proceed to delete.')
else:
    print(f'File {file_path} does not exist, skip deletion.')
4.执行删除操作

如果文件存在,您可以使用 os.remove() 函数来删除它。

python 复制代码
try:
    os.remove(file_path)
    print(f'File {file_path} deleted successfully.')
except OSError as e:
    print(f'Error occurred: {e}')
5. 完整的脚本示例
python 复制代码
import os
# 指定要删除的文件的目录
file_path = 'path_to_file'
# 检查文件是否存在
if os.path.isfile(file_path):
    print(f'File {file_path} exists, proceed to delete.')
else:
    print(f'File {file_path} does not exist, skip deletion.')
# 执行删除操作
try:
    os.remove(file_path)
    print(f'File {file_path} deleted successfully.')
except OSError as e:
    print(f'Error occurred: {e}')
6. 注意安全性和效率

在批量删除文件时,应确保:

不要同时进行多个删除操作,以避免潜在的竞争条件。

确保目标目录存在,避免在删除时创建不存在的目录。

考虑到操作系统对文件删除的操作限制,例如在 Windows 中,文件名不能超过 255 个字符,而在 Unix/Linux 中则没有这个限制。

通过以上步骤,您应该能够掌握如何使用Python删除文件。

3.3 创建文件

在Python中,创建文件是一个相对简单的操作。我们可以使用 os 库中的 os.open() 函数或者 with 语句来创建文件。以下是一个超详细的入门指南,介绍如何使用Python创建文件:

1. 导入必要的库

首先,您需要导入Python的 os 库,它提供了许多与操作系统交互的函数。

python 复制代码
import os
2. 准备文件路径

要创建文件,您需要知道要创建的文件的路径。

python 复制代码
file_path = 'path_to_file'
3. 检查文件路径是否存在

在创建文件之前,最好检查该文件路径是否存在,以避免覆盖其他文件。

python 复制代码
if not os.path.exists(file_path):
    print(f'File path {file_path} does not exist, proceed to create.')
else:
    print(f'File path {file_path} already exists, skip creation.')
4. 执行创建操作

如果文件路径不存在,您可以使用 os.open() 函数来创建文件。

python 复制代码
try:
    with open(file_path, 'w') as f:
        print(f'File {file_path} created successfully.')
except IOError as e:
    print(f'Error occurred: {e}')

这里,我们使用 with 语句来确保文件在操作完成后会被正确关闭。'w' 参数表示以写入模式打开文件,如果文件不存在,会创建一个新文件。

5. 完整的脚本示例
python 复制代码
import os
# 指定要创建的文件的目录
file_path = 'path_to_file'
# 检查文件路径是否存在
if not os.path.exists(file_path):
    print(f'File path {file_path} does not exist, proceed to create.')
else:
    print(f'File path {file_path} already exists, skip creation.')
# 执行创建操作
try:
    with open(file_path, 'w') as f:
        print(f'File {file_path} created successfully.')
except IOError as e:
    print(f'Error occurred: {e}')
6. 注意安全性和效率

在创建文件时,应确保:

  • 拥有创建文件的足够权限。
  • 避免在内存不足的情况下创建大型文件。
    通过以上步骤,您应该能够掌握如何使用Python创建文件。

3.4 获取当前目录

在Python中,我们可以使用 os 库中的 os.getcwd() 函数来获取当前目录的路径。以下是一个示例:

python 复制代码
import os
current_directory = os.getcwd()
print(f'Current directory is: {current_directory}')

4 示例

4.1 目录.txt自动清洗

  1. 需要在二级标题所在行最前面空4个格子,一级标题不用
  2. 需要在章和节字的后面加上一个空格
  3. 需要在页码前面加上=>符号
python 复制代码
# 获取桌面路径
import os
import re

desktop_path = os.path.join(os.path.expanduser("~"), "Desktop")

# 目标文件路径
file_path = os.path.join(desktop_path, "目录.txt")

# 打开文件并读取内容
with open(file_path, 'r', encoding='utf-8') as file:
    lines = file.readlines()

modified_lines = []
for line in lines:
    # 去除空格
    line = line.replace(" ", "")
    if len(line) == 1:
        continue
    # 使用正则表达式在'章'或'节'后面添加一个空格,仅在后面没有空格的情况下
    line = re.sub(r'(章|节)(?![ ])', r'\1 ', line)
    # 在小数点后添加空格
    line = re.sub(r'(\.\d)', r'\1 ', line)
    if '章' not in line:
        # 二级标题添加4个空格
        line = ' ' * 4 + line
    # 匹配并去除最外层的英文括号
    pattern_en = r'\(([\d\s]+)\)'
    line = re.sub(pattern_en, r'\1', line)
    # 匹配并去除最外层的中文括号及其内部内容(包括除数字和空格以外的字符)
    pattern = r'(([^)]+))'
    line = re.sub(pattern, r'\1', line)
    # 确保每行只有一个 =>
    if '=>' not in line:
        # 在页码数字前添加 =>(只在行尾)
        line = re.sub(r'(\d+)$', r'=>\1', line)
    # 去除中文汉字和'=>整体符号左边的冗余符号
    pattern = r'([\u4e00-\u9fff]+)[^\w\s]+=>'
    line = re.sub(pattern, r'\1=>', line)
    modified_lines.append(line)
# 将修改后的内容写回文件
with open(file_path, 'w', encoding='utf-8') as file:
    file.writelines(modified_lines)

# 读取文件内容
with open(file_path, 'r', encoding='utf-8') as file:
    content = file.read()
    print(content)

4.2 批量修改文件夹下的文件命名

D:\231127\试卷\00159高级财务会计 目录下的所有图片命名中有_,确保_前面的字符串为00159231127

你可以使用 Pythonos 模块来实现对文件名的批量修改,结合字符串操作来确保文件名中的规定格式。以下是一个示例代码:

python 复制代码
import os

# 指定目录路径
directory_path = r'D:\231127\试卷\00159高级财务会计'

# 获取目录下所有文件名
files = os.listdir(directory_path)

# 遍历文件
for file_name in files:
    # 构建完整的文件路径
    file_path = os.path.join(directory_path, file_name)

    # 检查文件是否是图片文件,并且文件名中包含下划线
    if file_name.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')) and '_' in file_name:
        # 分割文件名,以下划线为界
        parts = file_name.split('_')

        # 确保分割后的第一部分为'00159231127'
        if parts[0] != '00159231127':
            # 构建新文件名
            new_file_name = '00159231127' + '_' + '_'.join(parts[1:])

            # 构建新文件路径
            new_file_path = os.path.join(directory_path, new_file_name)

            # 重命名文件
            os.rename(file_path, new_file_path)
            print(f'Renamed: {file_name} -> {new_file_name}')

在上述代码中:

1.使用 os.listdir 获取目录下的所有文件名,然后遍历这些文件名。

2.通过 os.path.join 构建完整的文件路径,确保路径的正确性。

3.检查文件是否是图片文件(以 .png, .jpg, .jpeg, .gif 结尾的文件),并且文件名中包含下划线。

4.使用 split('_') 分割文件名,确保分割后的第一部分为'00159231127'。

5.构建新文件名,并使用 os.rename 来重命名文件。

4.3 检测同级目录下是否存在同名文件夹

检测 D:\231127\试卷 下的文件夹命名前5位是否存在一样的,如果一样打印出来

你可以使用 Python 来检查指定目录下的文件夹,查找前5位命名相同的文件夹。以下是一个示例代码:

python 复制代码
import os

# 指定目录路径
directory_path = r'D:\231127\试卷'

# 获取目录下所有文件夹名
folders = [folder for folder in os.listdir(directory_path) if os.path.isdir(os.path.join(directory_path, folder))]

# 创建一个空字典,用于存储前5位相同的文件夹名
same_prefix_folders = {}

# 遍历文件夹
for folder in folders:
    # 获取前5位文件夹名
    prefix = folder[:5]

    # 检查前5位文件夹名是否已经在字典中
    if prefix in same_prefix_folders:
        # 将文件夹名添加到对应的键值中
        same_prefix_folders[prefix].append(folder)
    else:
        # 若字典中没有该前5位文件夹名,则创建键值对
        same_prefix_folders[prefix] = [folder]

# 输出前5位相同的文件夹名
for prefix, folders in same_prefix_folders.items():
    if len(folders) > 1:
        print(f"前5位为 '{prefix}' 的文件夹有以下重复命名:")
        print(', '.join(folders))

这段代码执行以下操作:

1.使用 os.listdir 获取指定目录下的所有文件夹名。

2.然后遍历这些文件夹名,提取前 5 位名称,并将具有相同前缀的文件夹放入一个字典中。

3.最后打印出前 5 位相同的文件夹名。

原文:

python超详细基础文件操作【建议收藏】_python程序设计基础+文件操作-CSDN博客

扩展:

爬虫经常需要将数据保存到excel表格中,除了普通的csv模块以外,还有其他方式

1.pandas

# 使用 pandas 库:
 
import pandas as pd
 
filename = './data/test.csv'
 
df = pd.read_csv(filename)
 
print('df的数据类型为:{}'.format(type(df)))
print(df)

2.openpyxl

def write_to_excel(house_data, file_path):
    wb = openpyxl.Workbook()
    sheet = wb.active
    # 创建sheet表格,添加列描述
    sheet['A1'] = '房屋信息描述'
    sheet['B1'] = '房屋单价(每平方)'
    sheet['C1'] = '房屋总价(万)'
    # 循环遍历data列表中的元素,row是当前行数(从2开始),row_data是当前元素的内容
    # enumerate函数用于同时获取列表中的元素和它们的索引值
    for row, row_data in enumerate(house_data, start=2):
        # 分别将当前列表的值写到对应的行
        sheet[f'A{row}'] = row_data[0]
        sheet[f'B{row}'] = row_data[1]
        sheet[f'C{row}'] = row_data[2]
    wb.save(file_path)
    print(f"链家二手房信息表创建完成,保存路径为: {file_path}")
相关推荐
进击的六角龙1 小时前
Python中处理Excel的基本概念(如工作簿、工作表等)
开发语言·python·excel
一只爱好编程的程序猿1 小时前
Java后台生成指定路径下创建指定名称的文件
java·python·数据下载
Aniay_ivy1 小时前
深入探索 Java 8 Stream 流:高效操作与应用场景
java·开发语言·python
gonghw4032 小时前
DearPyGui学习
python·gui
向阳12182 小时前
Bert快速入门
人工智能·python·自然语言处理·bert
engchina2 小时前
Neo4j 和 Python 初学者指南:如何使用可选关系匹配优化 Cypher 查询
数据库·python·neo4j
兆。2 小时前
掌握 PyQt5:从零开始的桌面应用开发
开发语言·爬虫·python·qt
南宫理的日知录2 小时前
99、Python并发编程:多线程的问题、临界资源以及同步机制
开发语言·python·学习·编程学习
coberup2 小时前
django Forbidden (403)错误解决方法
python·django·403错误
龙哥说跨境3 小时前
如何利用指纹浏览器爬虫绕过Cloudflare的防护?
服务器·网络·python·网络爬虫