在python中使用Pandas对实时Websocket进行重采样

import pandas as pd
import websocket
import json

# 假设这是你的WebSocket消息处理函数
def on_message(ws, message):
    data = json.loads(message)
    # 假设data是一个包含时间戳和值的字典
    process_data(data)

# 数据处理函数
def process_data(data):
    global df
    # 将新数据添加到DataFrame中
    new_row = pd.DataFrame([data])
    df = pd.concat([df, new_row], ignore_index=True)
    
    # 设置时间戳为索引
    df.set_index('timestamp', inplace=True)
    
    # 重采样，例如将数据从秒级别转为分钟级别
    resampled_df = df.resample('T').mean()
    
    print(resampled_df)

# 初始化DataFrame
df = pd.DataFrame(columns=['timestamp', 'value'])

# WebSocket连接设置
websocket.enableTrace(True)
ws = websocket.WebSocketApp("ws://your-websocket-url",
                            on_message = on_message)
ws.run_forever()