pythonで配列の要素をcsvファイルに改行して出力したい

前提・実現したいこと

マウスで取得した座標を、配列に追加してそれをcsvファイルに出力しようとしているのですが、全て一行目に出力されます。理想としては、y座標まで出力するごとに改行するようにしたいです。

100 100
200 200
300 300 ・・・

というイメージなのですが、現在は、

100 100 200 200 300 300 ・・・

という感じです。

該当のソースコード

python
1import cv2
2import numpy as np
3import csv
4  
5    #メインループ
6    def run(self):
7        n = 1
8        
9        #最初のフレームの処理
10        self.frame = cv2.imread("IMG_1.tiff")
11        self.gray_prev = cv2.cvtColor(self.frame, cv2.COLOR_BGR2GRAY)
12        #self.gray_prev = self.frame
13        self.gray_next = cv2.cvtColor(self.frame,cv2.COLOR_BGR2GRAY)
14        #self.gray_next = self.frame
15        
16        #特徴点10点を指定
17        cv2.imshow("input window",self.frame)
18             
19        #コールバックの設定
20        mouseData = mouseParam("input window")
21    
22        while 1:
23            cv2.waitKey(10)
24            #左クリックがあったら表示
25            if mouseData.getEvent() == cv2.EVENT_LBUTTONDOWN:
26                mouseData.getX()
27                mouseData.getY()
28                mouseData.getPos()
29                print(mouseData.getX())
30                print(mouseData.getY())
31                self.onMouse(mouseData.getX(),mouseData.getY(),mouseData.getPos())               
32                
33            #右クリックがあったら終了
34            elif mouseData.getEvent() == cv2.EVENT_RBUTTONDBLCLK:
35                break;
36            
37            
38            #特徴点描画と10点の特徴点を取得
39            if self.features is not None:
40                if len(self.features) >= 10:
41                
42                    #オプティカルフローの計算
43                    features_prev = self.features
44                    #グレースケールに変換
45                    self.gray_next = cv2.cvtColor(self.frame,cv2.COLOR_BGR2GRAY)
46                    self.features, self.status, err = cv2.calcOpticalFlowPyrLK( \
47                                                        self.gray_prev, \
48                                                        self.gray_next, \
49                                                        features_prev, \
50                                                        None, \
51                                                        winSize = (10, 10), \
52                                                        maxLevel = 3, \
53                                                        criteria = CRITERIA, \
54                                                        flags = 0)
55                    #有効な特徴点のみ残す
56                    self.refreshFeatures()
57                    
58                    if len(self.features) >= 10:
59                        #最初の特徴点を格納しておく
60                        self.status_src = self.status
61                        self.features_src = self.features
62                        print("特徴点10個取得出来ました") 
63                        break;   
64                        
65        #print(self.features_src)
66        #print(self.features)
67        
68        #画像毎の特徴点を抽出し、追跡する
69        for frame in range(2,FRAME):
70           
71            #グレースケールに変換
72            self.gray_next = cv2.cvtColor(self.frame,cv2.COLOR_BGR2GRAY)
73            #self.gray_next = self.frame
74            
75            #特徴点が登録されている場合にOpticalFlowを計算
76            if self.features is not None:
77                #オプティカルフローの計算
78                features_prev = self.features
79                self.features, self.status, err = cv2.calcOpticalFlowPyrLK( \
80                                                    self.gray_prev, \
81                                                    self.gray_next, \
82                                                    features_prev, \
83                                                    None, \
84                                                    winSize = (10, 10), \
85                                                    maxLevel = 3, \
86                                                    criteria = CRITERIA, \
87                                                    flags = 0)
88                #有効な特徴点のみ残す
89                self.refreshFeatures()
90                
91                #フレームに有効な特徴点を描画
92                if self.features is not None:
93                    for features in self.features:
94                        cv2.circle(self.frame,(features[0][0],features[0][1]),2,(15,241,255),-1,8,0)
95                
96                #座標を保存する
97                if len(self.features) >= 10:
98
99                    #csvファイルに保存
100                    n += 1
101                    with open('frame%d'%n+'.csv','w',newline="") as f: 
102                        csvWriter = csv.writer(f,lineterminator='\n')
103                        listData = []
104                        for lists in self.features:
105                            listData.append(lists[0][0])
106                            listData.append(lists[0][1])
107                        csvWriter.writerow(listData)
108                        f.close()
109                        
110            #表示
111            cv2.imshow("motion",self.frame)
112            
113            #次のループ処理の準備
114            self.gray_prev = self.gray_next
115                
116            #画像を読み込む
117            self.frame = cv2.imread("IMG_"+str(frame)+".tiff")
118            
119            #インターバル
120            key = cv2.waitKey(self.interval)
121            
122        #終了処理
123        cv2.destroyAllWindows()

補足情報

・Windows7 32bit
・Python ver.3.7.1

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

回答1件

ベストアンサー

Python

listData = []
for lists in self.features:
listData.append(lists[0][0])
listData.append(lists[0][1])
csvWriter.writerow(listData)

未検証ですが、次のように書けば良いと思います。

Python
1for lists in self.features:
2    csvWriter.writerow([lists[0][0], lists[0][1]])

投稿2019/01/07 08:22

LouiS0616

総合スコア35660

chachamaru0421

2019/01/08 02:31

できました！ご回答くださり、ありがとうございました。

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

あなたの回答

tips

プレビュー

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

質問の解決につながる回答をしましょう。サンプルコードなど、より具体的な説明があると質問者の理解の助けになります。また、読む側のことを考えた、分かりやすい文章を心がけましょう。

15分調べてもわからないことは
teratailで質問しよう！

ただいまの回答率
85.48%

質問をまとめることで
思考を整理して素早く解決

テンプレート機能で
簡単に質問をまとめる

質問する

質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう！