Python从结果提取值：如何提取timestamp和value并写入CSV

没问题，我来帮你一步步搞定这个需求！不管你的脚本是直接输出内容，还是需要捕获外部命令的输出，我们都能用Python轻松实现目标。

第一步：先明确你的输出格式 #

首先得确定输出行的具体样式，比如常见的可能是这样：

timestamp: 2024-05-20 14:30:00, value: 156.78
或者
[2024-05-20T14:30:00] - 当前value: 156.78

我会用正则表达式做通用匹配，不管格式怎么变，只要能识别timestamp和value的特征就能适配。

第二步：两种常见场景的解决方案 #

场景1：修改你自己的Python脚本，直接收集数据（最高效） #

如果这是你自己写的脚本，完全不用等输出后再提取——在生成输出的同时就把timestamp和value存起来，最后直接写入CSV，这是最省事的方法：

import csv

# 替换成你脚本里实际的业务逻辑
data_list = []
for i in range(10):  # 示例循环，模拟生成数据的过程
    # 这里替换成你脚本中生成timestamp和value的代码
    timestamp = f"2024-05-20 14:{30+i}:00"
    value = 150.0 + i * 2.5
    
    # 保留你原来的输出逻辑
    print(f"timestamp: {timestamp}, value: {value}")
    # 同时收集数据
    data_list.append({"timestamp": timestamp, "value": value})

# 写入CSV文件
with open("result.csv", "w", newline="", encoding="utf-8") as csv_file:
    # 定义表头
    field_names = ["timestamp", "value"]
    writer = csv.DictWriter(csv_file, fieldnames=field_names)
    
    writer.writeheader()  # 写入表头行
    writer.writerows(data_list)  # 批量写入所有数据

场景2：从已有的输出内容（日志/命令输出）提取 #

如果是已经输出的内容（比如存在日志文件里，或者需要捕获其他命令的输出），我们可以读取每行内容，用正则提取需要的字段：

import csv
import re

# 假设你已经拿到了输出的行列表，比如从文件读取或subprocess捕获
# 这里用示例数据模拟实际输出
output_lines = [
    "timestamp: 2024-05-20 14:30:00, value: 156.78",
    "timestamp: 2024-05-20 14:31:00, value: 159.28",
    "timestamp: 2024-05-20 14:32:00, value: 161.78"
]

# 定义正则匹配规则：适配大多数timestamp和value的格式，可根据实际调整
pattern = re.compile(r'timestamp:\s*(.+?)\s*,?\s*value:\s*(\d+\.?\d*)')

extracted_data = []
for line in output_lines:
    match_result = pattern.search(line)
    if match_result:
        # 提取并整理数据
        timestamp = match_result.group(1).strip()
        value = float(match_result.group(2))  # 转成数值类型方便后续处理
        extracted_data.append({"timestamp": timestamp, "value": value})

# 写入CSV
with open("extracted_result.csv", "w", newline="", encoding="utf-8") as csv_file:
    writer = csv.DictWriter(csv_file, fieldnames=["timestamp", "value"])
    writer.writeheader()
    writer.writerows(extracted_data)

小提示 #

• 如果你的输出格式和示例不一样，只需要调整正则表达式pattern即可。比如如果timestamp是方括号括起来的，就改成r'\[(.+?)\].*value:\s*(\d+\.?\d*)'
• 用csv模块比直接写文本更安全，它会自动处理逗号、引号等特殊字符
• 如果数据量很大，用pandas会更便捷，比如pd.DataFrame(extracted_data).to_csv("result.csv", index=False)，不过需要先安装pandas：pip install pandas

内容的提问来源于stack exchange，提问作者Nataly Muñoz