Question

我想将从mosquitto broker收到的数据保存为csv文件。以下是mqtt_subscribe.py

的脚本

import paho.mqtt.client as mqtt
import pandas as pd

def on_connect(client, userdata, rc):
    print("Connected with result code "+str(rc))
    client.subscribe("test")

def on_message(client, userdata, msg):
    print str(msg.payload)
    datas = map(int, msg.payload)
    df = pd.DataFrame(data=datas, columns=['numbers'])

    f = open("test.csv", 'a')
    df.to_csv(f)
    f.close()

client = mqtt.Client()
client.on_connect = on_connect
client.on_message = on_message

client.connect("localhost", 1883, 60)

client.loop_forever()

此脚本将打印从代理接收的随机数/数据（直到我手动停止脚本），如下所示（示例）

Connected with result code 0
4
7
7

我希望将此结果写入CSV文件中，如此

,numbers
0,4
1,7
2,7

但我得到了这个

,numbers
0,4
,numbers
0,7
,numbers
0,7

我在这里遗漏了什么吗？我相信这是因为方法on_message会一直用列覆盖数据框，但我不知道除了on_message方法之外我应该在哪里初始化数据框。

提前谢谢。

Answer 1

您可以使用类似的方法将panda dataframe df附加到test.csv而不包含标题：

with open('test.csv', 'a') as csv_file:
    df.to_csv(csv_file, header=False)

使用＆＃39;使用open＆＃39;帮助我们不必担心明确关闭文件

如何用pandas将新行附加到csv文件？

1 个答案: