Pythonでcsvファイルを出力する際に，列ごとに名前を入力したい．

Question

実現したいこと

paraviewでのデータ整理のために出力したcsvファイルの列に名前をつけたい

前提

Pythonで輝度値を出力しparaviewでの解析を試みています。
輝度をcsvファイルに列ごとに出力することはできましたが，列の名前が入力されていないので，paraview上で時々データが読み込まれないことがあります．以下のように名前の下に値があるファイルを出力したいですが，x,y,valueの部分がないのが現状です．

x y value
0 0 5
1 2 3

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

該当のソースコード

import csv
from PIL import Image
import numpy as np
import pandas as pd

# 画像読込

for i in range(30, 91, 30):
   # 画像読込                             
   img = Image.open("img_00"+str(i)+".png")

   # モノクロ画像へ変換
   img = img.convert("L")
    
   width=720
   height=480
   ########################################################
   # 輝度値の取得、CSVファイルに保存

   # 画像の輝度値をlistで取得
   data = list(img.getdata())

   # 輝度値をCSVファイルで保存
   with open('image_data'+str(i)+'.csv', 'w+', newline='') as csvfile:
       spamwriter  = csv.writer(csvfile)

       # 画像データを一行ごと書き込み
       x = 0
       for y in range(height):
           for x in range(width):
               spamwriter.writerow([x, y, data[y*width+x]])

試したこと

今は手動で行を挿入して，名前を入力していますが，できれば自動にしたいです．

Answer 1

writerow() でヘッダ行を出力すればよいかと思います。

   # 輝度値をCSVファイルで保存
   with open('image_data'+str(i)+'.csv', 'w+', newline='') as csvfile:
       spamwriter  = csv.writer(csvfile)

       # ヘッダ行を出力
       spamwriter.writerow(('x', 'y', 'value'))

       # 画像データを一行ごと書き込み
       x = 0
       for y in range(height):
           for x in range(width):
               spamwriter.writerow([x, y, data[y*width+x]])

Comments

[afro087]

解決しました．ありがとうございます！！