python 读其他文件的函数

原创

mob649e815f0f18 2023-08-16 09:13:50 ©著作权

文章标签 CSV Python 文本文件 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob649e815f0f18的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python读其他文件的函数

在Python中，我们可以使用各种库和函数来读取和处理不同类型的文件，如文本文件、CSV文件、Excel文件等。本文将介绍一些常用的函数和方法，以及如何使用它们来读取其他文件。

1. 读取文本文件

在Python中，我们可以使用内置的open()函数来打开和读取文本文件。open()函数接受两个参数：文件路径和打开模式。常见的打开模式有：

'r'：只读模式，用于读取文件内容。
'w'：写入模式，用于写入文件内容。如果文件不存在，则创建新文件；如果文件已存在，则清空文件内容。
'a'：追加模式，用于向文件末尾添加内容。如果文件不存在，则创建新文件。
'x'：创建模式，用于创建新文件。如果文件已存在，则抛出FileExistsError异常。

以下是一个示例，演示如何使用open()函数读取文本文件：

def read_text_file(file_path):
    try:
        with open(file_path, 'r') as f:
            content = f.read()
            return content
    except FileNotFoundError:
        print("File not found.")
        return None

上述代码定义了一个read_text_file()函数，它接受一个文件路径作为参数，并尝试打开和读取该文件。如果文件不存在，则会捕获FileNotFoundError异常，并打印出错误信息。

2. 读取CSV文件

CSV（逗号分隔值）是一种常见的文件格式，用于存储表格数据。Python中有多个库可以用来读取和处理CSV文件，如csv、pandas等。

在使用csv库读取CSV文件时，我们可以使用csv.reader对象来逐行读取文件内容。以下是一个示例，演示如何使用csv库读取CSV文件：

import csv

def read_csv_file(file_path):
    try:
        with open(file_path, 'r') as f:
            reader = csv.reader(f)
            data = []
            for row in reader:
                data.append(row)
            return data
    except FileNotFoundError:
        print("File not found.")
        return None

上述代码定义了一个read_csv_file()函数，它接受一个文件路径作为参数，并尝试打开和读取该文件。使用csv.reader()方法逐行读取文件内容，并将每一行存储为一个列表，最后返回包含所有行的列表。

3. 读取Excel文件

Python中的pandas库提供了强大的功能来读取和处理Excel文件。我们可以使用pandas库中的read_excel()函数来读取Excel文件。以下是一个示例，演示如何使用pandas库读取Excel文件：

import pandas as pd

def read_excel_file(file_path):
    try:
        data = pd.read_excel(file_path)
        return data
    except FileNotFoundError:
        print("File not found.")
        return None

上述代码定义了一个read_excel_file()函数，它接受一个文件路径作为参数，并尝试使用pd.read_excel()函数读取Excel文件。返回的数据是一个pandas的DataFrame对象，可以方便地进行数据处理和分析。

类图

下面是一个展示以上函数的类图示例：

classDiagram
    class FileReader {
        - file_path : str
        + __init__(file_path: str)
        + read() : str|list|DataFrame|None
    }
    class TextFileReader {
        + read() : str
    }
    class CSVFileReader {
        + read() : list
    }
    class ExcelFileReader {
        + read() : DataFrame
    }
    FileReader <|-- TextFileReader
    FileReader <|-- CSVFileReader
    FileReader <|-- ExcelFileReader