如何使用 Pandas DataFrame 中的特定值更新文本文件中的特定位置

霞舞

发布时间：2025-08-17 19:22:01

793人浏览过

来源于php中文网

原创

如何使用 pandas dataframe 中的特定值更新文本文件中的特定位置

本文介绍如何使用 Python 的 Pandas 库从 DataFrame 中提取特定值，并将其写入文本文件中的特定位置，例如在特定标识符之后。重点在于利用正则表达式的强大功能，在复杂格式的文本文件中定位并替换目标数据，从而实现高效、灵活的数据更新。

使用 Pandas 和正则表达式更新文本文件

以下步骤展示了如何读取包含特定格式条目的文本文件，从 Pandas DataFrame 中选择一行，并将 DataFrame 中的 i、j 和 k 值替换到文本文件中与特定标识符关联的相应位置。

1. 准备工作

首先，确保你已经安装了 Pandas 和 re 模块（Python 的正则表达式模块）。Pandas 通常已经预装，如果没有，可以使用 pip install pandas 进行安装。re 模块是 Python 的内置模块，无需额外安装。

百度AI搜

百度全新AI搜索引擎

下载

2. 代码实现

import re
import pandas as pd

# 模拟 DataFrame (实际情况是从文件读取)
data = {'i': ['unit1', 1000, -3000, -2000, 900],
        'j': ['unit2', 100, 200, 90, 40],
        'k': ['unit3', 84, 60, 195, 209]}
df = pd.DataFrame(data)
df = df.drop(0) # 删除第一行 'unit' 行
df = df.reset_index(drop=True) # 重置索引

idx = 2 # 要使用的 DataFrame 行的索引（从0开始）
to_replace = "B" # 要替换的条目的标识符

# 假设 input_file.txt 包含以下内容：
# A       first = 4 | 1_3_5_4        Name1
# labelToSkip
# i = 1000000 j = -3 k = -15
# end
#
# B       first = 4 | 9_2_2_4        Name2
# labelToSkip
# i = 150000 j = -3 k = -20
# end

with open("input_file.txt", "r") as f_in:
    file_string = f_in.read()

    i, j, k = df.loc[idx, ["i", "j", "k"]].astype(int) # 确保值为整数

    file_string = re.sub(
        rf"^({to_replace}\s.*?)i = \S+ j = \S+ k = \S+",
        f"\g<1>i = {i} j = {j} k = {k}",
        file_string,
        flags=re.M | re.S,
    )

with open("output_file.txt", "w") as f_out:
    f_out.write(file_string)

3. 代码解释

导入必要的库: 导入 re (正则表达式) 和 pandas。
定义变量:
- idx: 指定要从 DataFrame 中提取数据的行索引。
- to_replace: 指定要修改的条目的标识符 (例如 "B")。
读取文件内容: 使用 open() 函数以读取模式打开 input_file.txt，并使用 f_in.read() 将整个文件内容读取到字符串变量 file_string 中。
提取 DataFrame 值: 使用 df.loc[idx, ["i", "j", "k"]] 从 DataFrame 中选择指定行和列的值。.astype(int) 将这些值转换为整数类型，这对于确保替换后的值类型正确非常重要。
使用正则表达式替换:
- re.sub() 函数执行替换操作。
- rf"^({to_replace}\s.*?)i = \S+ j = \S+ k = \S+": 这是正则表达式模式。
  - ^: 匹配行的开头。
  - ({to_replace}\s.*?): 匹配以 to_replace 变量（例如 "B"）开头的行，并捕获该行直到 i = 的部分。 \s 匹配空白字符，.*? 匹配任意字符（非贪婪模式）。
  - i = \S+ j = \S+ k = \S+: 匹配 i =、j = 和 k = 及其后的非空白字符。 \S+ 匹配一个或多个非空白字符。
- f"\g<1>i = {i} j = {j} k = {k}": 这是替换字符串。
  - \g<1>: 引用正则表达式中第一个捕获组（即 ({to_replace}\s.*?) 匹配的内容）。
  - i = {i} j = {j} k = {k}: 将 DataFrame 中提取的 i、j 和 k 的值插入到字符串中。
- flags=re.M | re.S: 设置正则表达式的标志。
  - re.M (或 re.MULTILINE): 使 ^ 和 $ 匹配每一行的开头和结尾，而不仅仅是整个字符串的开头和结尾。
  - re.S (或 re.DOTALL): 使 . 匹配任何字符，包括换行符。
写入更新后的内容: 使用 open() 函数以写入模式打开 output_file.txt，并使用 f_out.write(file_string) 将修改后的字符串 file_string 写入到文件中。

4. 注意事项

文件路径: 确保 input_file.txt 和 output_file.txt 的路径是正确的。
数据类型: 确保 DataFrame 中的数据类型与文本文件中期望的类型一致。在上面的示例中，我们使用 .astype(int) 将 DataFrame 中的值转换为整数。
正则表达式: 正则表达式是此解决方案的关键。仔细构建正则表达式以确保它正确匹配要替换的文本。使用在线正则表达式测试工具来验证你的表达式。
错误处理: 在实际应用中，应该添加错误处理机制，例如检查文件是否存在，以及 DataFrame 中是否存在所需的行和列。
备份: 在修改文件之前，务必备份原始文件。