如何用Python遍历InfluxDB查询结果并提取指定字段值存入数组

我来帮你梳理下怎么处理InfluxDB返回的结果集，以及提取你需要的usage_user、usage_system等值存入数组的方法。

首先你拿到的ResultSet对象本质是一个类似字典的结构，键是(measurement名称, tags字典)的元组（如果没有tags就是None），值对应该measurement下的所有数据点列表。下面是几种常用的遍历和提取方式：

1. 直接遍历所有数据点（最常用） #

用get_points()方法可以直接获取所有查询到的数据点，如果你只想针对特定measurement，还可以传入measurement参数过滤：

from influxdb import InfluxDBClient

influx_host = "你的主机地址"
influx_port = 8086
client = InfluxDBClient(host=influx_host, port=influx_port, database='db')
q = client.query("select * from cpu limit 1")

# 初始化存储数组
metrics_array = []

# 遍历cpu measurement下的数据点
for point in q.get_points(measurement='cpu'):
    # 提取需要的字段
    usage_user = point['usage_user']
    usage_system = point['usage_system']
    
    # 可以把单个值存入数组，或者构造字典存入
    # 方式1：存单个值
    metrics_array.append(usage_user)
    metrics_array.append(usage_system)
    
    # 方式2：存结构化数据（更推荐，方便后续使用）
    metrics_array.append({
        'time': point['time'],
        'usage_user': usage_user,
        'usage_system': usage_system
    })

# 查看结果
print(metrics_array)

2. 遍历ResultSet的键值对（获取measurement信息） #

如果你需要同时获取measurement名称和对应的数据点，可以直接遍历ResultSet的items()：

for (measurement, tags), points in q.items():
    print(f"当前Measurement: {measurement}，Tags: {tags}")
    # 遍历该measurement下的所有数据点
    for point in points:
        print(f"时间: {point['time']}，用户使用率: {point['usage_user']}")
        # 批量提取字段到数组
        usage_user_list = [p['usage_user'] for p in points]
        usage_system_list = [p['usage_system'] for p in points]
        print("所有usage_user值:", usage_user_list)

3. 针对你的查询结果的简化写法 #

因为你的查询是limit 1，只会返回一个数据点，也可以直接通过索引获取：

# 获取cpu measurement下的第一个数据点
points = list(q.get_points(measurement='cpu'))
if points:
    first_point = points[0]
    usage_user = first_point['usage_user']
    usage_system = first_point['usage_system']
    print(f"usage_user: {usage_user}, usage_system: {usage_system}")

总结一下，get_points()是最便捷的方式，适合大多数场景；如果需要区分不同的measurement或tags，就用items()遍历。

内容的提问来源于stack exchange，提问作者Cheetara