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

前提・実現したいこと

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

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

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

A
10  972  1023  972  1023 -1.216718
2     A     B    C    D         E
30  972  1023  972  995 -1.040281
4     A     B    C    D         E
50  972  1017  973  978  0.317648
6     A     B    C    D         E
70  973  1023  972  973 -0.133971
8     A     B    C    D         E
90  972  1023  972  972 -0.819937
10     A     B    C    D         E
110  973  1023  971  991 -0.398122
12     A     B    C    D         E
130  972  1023  971  979  0.290892
14     A     B    C    D         E
150  972  1020  970  971 -0.902272
16     A     B    C    D         E
170  972  1020  973  976  0.106279
18     A     B    C    D         E
190  972  1023  972  962 -0.090337
20
21
22
23##

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)})

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

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

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

行動規範の内容に同意します

回答3件

ベストアンサー

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

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

Python
1import pandas as pd
2import numpy as np
3import serial
4
5def main():
6    ser = serial.Serial("COM3")  # COMポート(Arduino接続)
7
8    df = pd.DataFrame(columns=["A","B","C","D"])
9
10    num = 0
11    while num < 10:
12
13        data = ser.readline().rstrip()  # \nまで読み込む(\nは削除
14        data = data.decode() #対話型だとこれが必要らしい
15        (xdeg, ydeg, zdeg, wdeg) =data.split(",")
16
17        new_data = pd.Series([xdeg, ydeg, zdeg, wdeg], index=df.columns)
18
19        df = df.append(new_data, ignore_index=True)
20
21        num = num + 1
22
23        print(df)
24
25    # CSVファイルの作成
26    df.to_csv('output.csv')
27
28if __name__ == '__main__':
29    main()

投稿2018/09/02 07:39

magichan

総合スコア15898

meJ15

2018/09/03 04:27

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

行動規範の内容に同意します

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

投稿2018/09/02 07:29

y_waiwai

総合スコア88163

meJ15

2018/09/03 04:26

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

行動規範の内容に同意します

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

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

Python
1with open('out.csv', 'w') as fout:
2    print(*list('ABCDE'), sep=',', file=fout)
3
4    for _ in range(10):
5        data = ser.readline().rstrip()
6        data = data.decode()
7
8        print(
9            *data.split(','), np.random.randn(1),
10            sep=',', file=fout
11        )