使用 Pandas 高效关联多文件数据并提取特定信息

碧海醫心
发布: 2025-09-21 11:33:01
原创
938人浏览过

使用 Pandas 高效关联多文件数据并提取特定信息

本教程详细介绍了如何利用 Python 的 Pandas 库,高效地处理来自多个文本文件的关联数据。通过将文件内容加载到 DataFrame 中,并运用 merge 操作进行数据整合,实现 IP 地址、MAC 地址和端口信息的精确匹配与提取,最终生成结构化的输出结果,避免了传统文件处理的复杂性。

场景需求:多文件数据关联与信息提取

在日常的数据处理任务中,我们经常会遇到需要从多个关联文件中提取和整合信息的情况。例如,给定三个文件:

  • file1.txt 包含一系列待查询的 IP 地址。
  • file2.txt 包含 IP 地址到 MAC 地址的映射关系。
  • file3.txt 包含 MAC 地址到端口的映射关系。

我们的目标是,对于 file1.txt 中的每个 IP 地址,首先在 file2.txt 中找到对应的 MAC 地址,然后利用这个 MAC 地址在 file3.txt 中找到对应的端口,最终以 "IP 地址 MAC 地址 端口" 的格式输出所有匹配的结果。

传统的文件处理方法,如逐行读取、嵌套循环和字符串匹配,在处理这类关联数据时往往效率低下且代码复杂,尤其当文件较大时,性能问题会更加突出。

Pandas 解决方案概览

Python 的 Pandas 库提供了一套强大的数据结构(DataFrame)和数据分析工具,特别适用于处理表格型数据。利用 Pandas,我们可以将每个文件视为一个 DataFrame,然后通过类似于 SQL 的 merge(合并)操作,高效地将这些 DataFrame 关联起来,从而轻松实现数据的整合和查询。这种方法不仅代码简洁,而且在处理大量数据时具有显著的性能优势。

数据准备:加载文件至 DataFrame

首先,我们需要将每个文本文件的内容加载到 Pandas DataFrame 中。在实际应用中,通常使用 pd.read_csv() 或 pd.read_table() 等函数从文件中读取数据。对于本教程的示例,我们将直接通过 Python 字典创建 DataFrame,以确保代码的自包含性和可运行性。

假设我们的原始文件内容如下:

file1.txt (IP地址列表):

1.1.1.1
1.1.1.2
1.1.1.3
1.1.1.6
1.1.1.11
登录后复制

file2.txt (IP到MAC的映射):

Protocol  Address   Age (min)  Addr            Type   Interface
Internet  1.1.1.1         5    6026.aa11.1111  A      Ethernet1/49
Internet  1.1.1.2         -    0006.f2d2.2d2f  A      Vlan1
Internet  1.1.1.3         -    6026.aa33.3333  A      Vlan1
Internet  1.1.1.4         0    Incomplete      A
Internet  1.1.1.5         0    Incomplete      A
Internet  1.1.1.6         64   fa16.6edb.6666  A      Vlan1
Internet  1.1.1.11        23   fa16.7e7d.7777  A      Vlan1
登录后复制

file3.txt (MAC到端口的映射):

Unicast Entries
 vlan     mac address     type        protocols               port
---------+---------------+--------+---------------------+-------------------------
 1        6026.aa11.1111   static  ip,ipx,assigned,other Switch
 1        0006.f2d2.2d2f   dynamic ip,ipx,assigned,other Ethernet1/24
 1        6026.aa33.3333   dynamic ip,ipx,assigned,other Ethernet1/12
 1        fa16.6edb.6666   dynamic ip,ipx,assigned,other Ethernet1/8
 1        fa16.7e7d.7777   dynamic ip,ipx,assigned,other Ethernet1/10
登录后复制

对应的 DataFrame 创建代码如下:

微信 WeLM
微信 WeLM

WeLM不是一个直接的对话机器人,而是一个补全用户输入信息的生成模型。

微信 WeLM 33
查看详情 微信 WeLM
import pandas as pd

# 假设 file1.txt 只有一列IP地址,无表头
# 实际读取文件示例: df1 = pd.read_csv('file1.txt', header=None, names=['ipv4'])
df1 = pd.DataFrame({"ipv4":{"0":"1.1.1.1","1":"1.1.1.2","2":"1.1.1.3","3":"1.1.1.6","4":"1.1.1.11"}})

# 假设 file2.txt 有表头,并且是空格分隔
# 实际读取文件示例: df2 = pd.read_csv('file2.txt', delim_whitespace=True)
df2 = pd.DataFrame({
  "Protocol":{ "0":"Internet", "1":"Internet", "2":"Internet", "3":"Internet", "4":"Internet", "5":"Internet", "6":"Internet" },
  "Address":{ "0":"1.1.1.1", "1":"1.1.1.2", "2":"1.1.1.3", "3":"1.1.1.4", "4":"1.1.1.5", "5":"1.1.1.6", "6":"1.1.1.11" },
  "Age (min)":{ "0":"5", "1":"-", "2":"-", "3":"0", "4":"0", "5":"64", "6":"23" },
  "Addr":{ "0":"6026.aa11.1111", "1":"0006.f2d2.2d2f", "2":"6026.aa33.3333", "3":"Incomplete", "4":"Incomplete", "5":"fa16.6edb.6666", "6":"fa16.7e7d.7777" },
  "Type":{ "0":"A", "1":"A", "2":"A", "3":"A", "4":"A", "5":"A", "6":"A" },
  "Interface":{ "0":"Ethernet1/49", "1":"Vlan1", "2":"Vlan1", "3":None, "4":None, "5":"Vlan1", "6":"Vlan1" }
})

# 假设 file3.txt 有表头,并且是空格分隔
# 实际读取文件示例: df3 = pd.read_csv('file3.txt', delim_whitespace=True, skiprows=[1]) # skiprows跳过分隔线
df3 = pd.DataFrame({
    "vlan":{"0":1,"1":1,"2":1,"3":1,"4":1},
    "mac address":{"0":"6026.aa11.1111","1":"0006.f2d2.2d2f","2":"6026.aa33.3333","3":"fa16.6edb.6666","4":"fa16.7e7d.7777"},
    "type":{"0":"static","1":"dynamic","2":"dynamic","3":"dynamic","4":"dynamic"},
    "protocols":{"0":"ip,ipx,assigned,other","1":"ip,ipx,assigned,other","2":"ip,ipx,assigned,other","3":"ip,ipx,assigned,other","4":"ip,ipx,assigned,other"},
    "port":{"0":"Switch","1":" Ethernet1\/24","2":" Ethernet1\/12","3":" Ethernet1\/8","4":" Ethernet1\/10"}})
登录后复制

核心操作:使用 merge 函数整合数据

Pandas 的 merge 函数是实现 DataFrame 之间关联的核心工具。它类似于 SQL 中的 JOIN 操作,可以根据一个或多个共同的列将两个 DataFrame 合并。

在本例中,我们需要进行两次合并:

  1. 第一次合并:将 df1(包含IP地址)与 df2(包含IP到MAC的映射)合并。
    • df1 中的 ipv4 列与 df2 中的 Address 列是共同的关联键。
    • 我们使用 how="inner" 来确保只有在两个 DataFrame 中都存在的 IP 地址才会被保留。
  2. 第二次合并:将第一次合并的结果与 df3(包含MAC到端口的映射)合并。
    • 第一次合并结果中的 Addr 列(即MAC地址)与 df3 中的 mac address 列是共同的关联键。

合并操作如下:

# 第一次合并:根据IP地址关联 df1 和 df2
# left_on="ipv4" 指 df1 的关联列,right_on="Address" 指 df2 的关联列
merged_df_ip_mac = df1.merge(df2, how="inner", left_on="ipv4", right_on="Address")

# 第二次合并:根据MAC地址关联第一次合并的结果和 df3
# left_on="Addr" 指 merged_df_ip_mac 的关联列,right_on="mac address" 指 df3 的关联列
maindf = merged_df_ip_mac.merge(df3, how="inner", left_on="Addr", right_on="mac address")
登录后复制

通过这两次 inner 合并,maindf 中将只包含那些在所有三个文件中都能找到对应关系的 IP、MAC 和端口信息。

结果输出:提取并格式化所需信息

合并完成后,maindf 包含了所有我们需要的关联数据。现在,我们只需要从 maindf 中选择我们关心的列 (ipv4, Addr, port),并按照指定格式输出。

# 提取所需的列
result_df = maindf[["ipv4", "Addr", "port"]]

# 按照指定格式打印结果
print("期望输出:")
for index, row in result_df.iterrows():
    print(f"ip {row['ipv4']} addr {row['Addr']} port {row['port']}")
登录后复制

这将产生以下输出:

ip 1.1.1.1 addr 6026.aa11.1111 port Switch
ip 1.1.1.2 addr 0006.f2d2.2d2f port Ethernet1/24
ip 1.1.1.3 addr 6026.aa33.3333 port Ethernet1/12
ip 1.1.1.6 addr fa16.6edb.6666 port Ethernet1/8
ip 1.1.1.11 addr fa16.7e7d.7777 port Ethernet1/10
登录后复制

完整示例代码

以下是整合了所有步骤的完整 Python 代码:

import pandas as pd

# 1. 数据准备:加载文件至 DataFrame (此处为演示目的,直接创建DataFrame)
# 实际文件读取示例:
# df1 = pd.read_csv('file1.txt', header=None, names=['ipv4'])
# df2 = pd.read_csv('file2.txt', delim_whitespace=True)
# df3 = pd.read_csv('file3.txt', delim_whitespace=True, skiprows=[1]) # 假设需要跳过第二行分隔线

df1 = pd.DataFrame({"ipv4":{"0":"1.1.1.1","1":"1.1.1.2","2":"1.1.1.3","3":"1.1.1.6","4":"1.1.1.11"}})

df2 = pd.DataFrame({
  "Protocol":{ "0":"Internet", "1":"Internet", "2":"Internet", "3":"Internet", "4":"Internet", "5":"Internet", "6":"Internet" },
  "Address":{ "0":"1.1.1.1", "1":"1.1.1.2", "2":"1.1.1.3", "3":"1.1.1.4", "4":"1.1.1.5", "5":"1.1.1.6", "6":"1.1.1.11" },
  "Age (min)":{ "0":"5", "1":"-", "2":"-", "3":"0", "4":"0", "5":"64", "6":"23" },
  "Addr":{ "0":"6026.aa11.1111", "1":"0006.f2d2.2d2f", "2":"6026.aa33.3333", "3":"Incomplete", "4":"Incomplete", "5":"fa16.6edb.6666", "6":"fa16.7e7d.7777" },
  "Type":{ "0":"A", "1":"A", "2":"A", "3":"A", "4":"A", "5":"A", "6":"A" },
  "Interface":{ "0":"Ethernet1/49", "1":"Vlan1", "2":"Vlan1", "3":None, "4":None, "5":"Vlan1", "6":"Vlan1" }
})

df3 = pd.DataFrame({
    "vlan":{"0":1,"1":1,"2":1,"3":1,"4":1},
    "mac address":{"0":"6026.aa11.1111","1":"0006.f2d2.2d2f","2":"6026.aa33.3333","3":"fa16.6edb.6666","4":"fa16.7e7d.7777"},
    "type":{"0":"static","1":"dynamic","2":"dynamic","3":"dynamic","4":"dynamic"},
    "protocols":{"0":"ip,ipx,assigned,other","1":"ip,ipx,assigned,other","2":"ip,ipx,assigned,other","3":"ip,ipx,assigned,other","4":"ip,ipx,assigned,other"},
    "port":{"0":"Switch","1":" Ethernet1\/24","2":" Ethernet1\/12","3":" Ethernet1\/8","4":" Ethernet1\/10"}})

# 2. 核心操作:使用 merge 函数整合数据
# 第一次合并:df1 (ipv4) -> df2 (Address, Addr)
merged_df_ip_mac = df1.merge(df2, how="inner", left_on="ipv4", right_on="Address")

# 第二次合并:merged_df_ip_mac (Addr) -> df3 (mac address, port)
maindf = merged_df_ip_mac.merge(df3, how="inner", left_on="Addr", right_on="mac address")

# 3. 结果输出:提取并格式化所需信息
result_df = maindf[["ipv4", "Addr", "port"]]

print("最终匹配结果:")
for index, row in result_df.iterrows():
    print(f"ip {row['ipv4']} addr {row['Addr']} port {row['port']}")
登录后复制

注意事项

  1. 文件读取参数
    • pd.read_csv() 是读取 CSV 文件的常用函数,但它也可以处理其他分隔符的文件。对于空格或不规则分隔符的文件,可以使用 sep='s+' 或 delim_whitespace=True。
    • 如果文件没有表头,需要设置 header=None,并可以通过 names 参数手动指定列名。
    • 如果文件

以上就是使用 Pandas 高效关联多文件数据并提取特定信息的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号