python中如何读取excel两列

原创

mob64ca12f5c08e 2024-01-23 04:16:16 ©著作权

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

Python中如何读取Excel两列

在Python中，我们可以使用pandas库来读取Excel文件，并且只获取其中的两列数据。

安装pandas

首先，我们需要安装pandas库。可以使用以下命令来安装：

pip install pandas

读取Excel文件

接下来，我们可以使用pandas的read_excel函数来读取Excel文件。以下是一个示例代码：

import pandas as pd

# 读取Excel文件
df = pd.read_excel('file.xlsx')

# 获取两列数据
column1 = df['Column1']
column2 = df['Column2']

# 打印数据
print(column1)
print(column2)

在上面的代码中，我们首先使用read_excel函数读取名为file.xlsx的Excel文件。然后，我们可以通过指定列名称来获取这两列数据。在这个示例中，我们使用df['Column1']和df['Column2']来获取名为Column1和Column2的两列数据。最后，我们使用print函数来打印这两列数据。

完整代码示例

下面是一个完整的代码示例，演示如何读取Excel文件的两列数据并打印出来：

import pandas as pd

def read_excel_columns(filename, column1_name, column2_name):
    # 读取Excel文件
    df = pd.read_excel(filename)

    # 获取两列数据
    column1 = df[column1_name]
    column2 = df[column2_name]

    # 返回两列数据
    return column1, column2

# 测试代码
column1, column2 = read_excel_columns('file.xlsx', 'Column1', 'Column2')
print(column1)
print(column2)

在上面的代码中，我们定义了一个名为read_excel_columns的函数，它接受三个参数：filename表示Excel文件的名称，column1_name和column2_name表示要获取的两列数据的列名称。该函数会返回这两列数据。然后，我们调用这个函数并打印返回的结果。

状态图

下面是一个使用mermaid语法标识的状态图，展示了读取Excel两列的过程：

stateDiagram
    [*] --> 读取Excel文件
    读取Excel文件 --> 获取两列数据
    获取两列数据 --> 打印数据
    打印数据 --> [*]

在这个状态图中，我们首先开始于初始状态[*]，然后按顺序进行读取Excel文件、获取两列数据和打印数据的操作，最后回到初始状态[*]。

序列图

下面是一个使用mermaid语法标识的序列图，展示了读取Excel两列的过程：

sequenceDiagram
    participant 用户
    participant Python脚本
    participant pandas库
    participant Excel文件

    用户 ->> Python脚本: 指定Excel文件名和列名称
    Python脚本 ->> pandas库: 调用read_excel函数
    pandas库 ->> Excel文件: 读取Excel文件
    Excel文件 -->> pandas库: 返回数据
    pandas库 -->> Python脚本: 返回数据
    Python脚本 ->> 用户: 打印数据

在这个序列图中，我们有四个参与者：用户、Python脚本、pandas库和Excel文件。用户首先通过调用Python脚本并指定Excel文件名和列名称。然后，Python脚本调用pandas库的read_excel函数来读取Excel文件。Excel文件返回数据给pandas库，然后pandas库将数据返回给Python脚本。最后，Python脚本将数据打印给用户。