pythonで4つのセンサの情報をcsvファイルに保存することを目指しているのですが詰まっています。

Question

前提・実現したいこと

ここに質問の内容を詳しく書いてください。
シリアル通信によりpythonで4つのセンサの情報をcsvファイルに保存することを目指しています。
4つのセンサの情報をいろんなサイトを調べてjuptyterで確認できるようになったのですが、csvファイルに書き込む前で詰まっています。

下のように出力されてしまいます。
一行目をABCDとしたいのですが、毎回センサの値を読むたびにABCDが出てきてしまいます。
プログラムのどこを変えればよいのでしょうか？

発生している問題・エラーメッセージ

0  972  1023  972  1023 -1.216718
     A     B    C    D         E
0  972  1023  972  995 -1.040281
     A     B    C    D         E
0  972  1017  973  978  0.317648
     A     B    C    D         E
0  973  1023  972  973 -0.133971
     A     B    C    D         E
0  972  1023  972  972 -0.819937
     A     B    C    D         E
0  973  1023  971  991 -0.398122
     A     B    C    D         E
0  972  1023  971  979  0.290892
     A     B    C    D         E
0  972  1020  970  971 -0.902272
     A     B    C    D         E
0  972  1020  973  976  0.106279
     A     B    C    D         E
0  972  1023  972  962 -0.090337



## 

python

import numpy as np
import matplotlib.pyplot as plt
import pygame
from pygame.locals import *
import serial
import sys
import pandas as pd

def main():
    ser = serial.Serial("COM3")  # COMポート(Arduino接続)
    xdegs = [0]*100              # 温度格納
    ydegs = [0]*100              # 温度格納
    zdegs = [0]*100              # 温度格納
    wdegs = [0]*100              # 温度格納
    t = np.arange(0,100,1)
    num = 0 
    
    while num < 10:

        data = ser.readline().rstrip()  # \nまで読み込む(\nは削除
        data = data.decode() #対話型だとこれが必要らしい
        (xdeg, ydeg, zdeg, wdeg) =data.split(",")
        
        a = (xdeg,ydeg, zdeg, wdeg)
 
        
        df = pd.DataFrame({
                   "A" : xdeg,
                   "B" : ydeg,
                   "C" : zdeg,
                   "D" : wdeg,
                   "E" : np.random.randn(1)})

        
        num = num + 1

        print(df)

if __name__ == '__main__':
    main() 

試したこと

import pandas as pd
import numpy as np

df = pd.DataFrame({"A" : ['foo', 'bar', 'foo', 'bar','foo', 'bar', 'foo', 'foo'],
"B" : ['one', 'one', 'two', 'three','two', 'two', 'one', 'three'],
"C" : np.random.randn(8),
"D" : np.random.randn(8)})

df

となることがわかり、このように出力した後に、CSVファイルに書き込みたいです。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Answer 1

どのタイミングで csvファイルを生成するのかよくわからないのですが、

とりあえず、while ループの外で、空のDataFrameを作成し、 loopの中で DataFrame.append()を使ってデータフレームに行を追加していくと良いのではないでしょうか

import pandas as pd
import numpy as np
import serial

def main():
    ser = serial.Serial("COM3")  # COMポート(Arduino接続)

    df = pd.DataFrame(columns=["A","B","C","D"])

    num = 0
    while num < 10:

        data = ser.readline().rstrip()  # \nまで読み込む(\nは削除
        data = data.decode() #対話型だとこれが必要らしい
        (xdeg, ydeg, zdeg, wdeg) =data.split(",")

        new_data = pd.Series([xdeg, ydeg, zdeg, wdeg], index=df.columns)

        df = df.append(new_data, ignore_index=True)

        num = num + 1

        print(df)

    # CSVファイルの作成
    df.to_csv('output.csv')

if __name__ == '__main__':
    main()

Comments

[meJ15]

回答ありがとうございます。意図していた動作をしました。
大変助かりました。

Answer 2

別回答を。
最初にそのCSVファイルが存在するかチェックして、存在してなければ、ファイルを新規作成し、"A,B,C,D"を書き込み、クローズ。
その後、ワンセットのセンサの値が揃い次第、ファイルに追記モードでオープンして値を書き込んでクローズ、と繰り返す

Comments

[meJ15]

回答ありがとうございます。
その考え方があまりなかったです。

Answer 3

毎回データフレームを作っているからです。
単にCSVにしたいだけなら、適当にファイル開いて直接書き込んだ方が楽かと。

以下、未検証ですが参考までに。

with open('out.csv', 'w') as fout:
    print(*list('ABCDE'), sep=',', file=fout)

    for _ in range(10):
        data = ser.readline().rstrip()
        data = data.decode()

        print(
            *data.split(','), np.random.randn(1),
            sep=',', file=fout
        )      

Arduinoは触ったことがないので、とんちんかんなことを書いていたらごめんなさい。

---

データフレームに結果を格納したいなら、毎周データを継ぎ足していって最後にまとめて出力します。

Comments

[meJ15]

回答ありがとうございます。
継ぎ足すという考え方がすごく参考になりました。