CSVファイルのデータの一部を抜き出してExcelファイルに出力する。

Question

前提・実現したいこと

業務効率化のためにpythonでエクセルを操作することを学んでいます。
CSVファイルで出力された測定データの値を抜き出して
１つのエクセルファイルに書き出したいです。
CSVファイルすべてに記載された値をエクセルにそのまま書き出すことだけはできたのですが
・範囲を指定して書き出すこと
・2列で記載されたデータを一列にしてエクセルファイルに書き出すこと
いろいろサイトを調べましたがこの２つが実現できませんでした。
ご回答のほど、よろしくお願い致します。

該当のソースコード

#CSVファイルをそのままエクセルに書き出すコード

import openpyxl
import csv

wb = openpyxl.Workbook()
ws = wb.active
ws.title = "new_data"

with open("..\data\sound_data\data1.csv") as f:
    r = csv.reader(f, delimiter=',')
    for row in r:
        ws.append(row)

wb.save("..\data\sound_data\output_data.xlsx")

試したこと

データをリストに格納してみたりしましたが
上手くいきませんでした。

Answer 1

csv パッケージでデータを読み込んで、縦に結合する処理はこんな感じとなります。
（１列目と２列目のデータを別のリストに格納し結合）
データの書き出しはセル指定で行っております。

import openpyxl
import csv

# データ読み込み部
with open("..\data\sound_data\data1.csv", encoding='SHIFT-JIS') as f:
    col1, col2 = [], []
    r = csv.reader(f, delimiter=',')
    for row in r:
        col1.append(row[0])
        col2.append(row[1])
    # １列目と２列目のデータを結合
    data = col1[1:]
    data.extend(col2[1:])

# データ書き込み部
wb = openpyxl.Workbook()
ws = wb.active
ws.title = "new_data"
for i,d in enumerate(data, start=2):
    # セルを指定してデータを書き出す（一応データを数値変換してます）
    ws.cell(row=i, column=1, value=int(d))
wb.save("..\data\sound_data\output_data.xlsx")

ただ、この手の処理を行いたいのであれば、個人的にはpandasをお勧めしたいところです。
pandasならば、csvの読み込み、列同士の結合、excelへの書き出しが全てサポートされてますので
下記のように数行で書けます。

import pandas as pd

df = pd.read_csv('..\data\sound_data\data1.csv', encoding='SHIFT-JIS')
ser = pd.concat([df['値1'], df['値2']], ignore_index=True).rename('')
ser.to_excel('..\data\sound_data\output_data.xlsx', index=None, sheet_name='new_data', encoding='SHIFT-JIS')

Comments

[退会済みユーザー]

参考にさせていただきます。ありがとうございます。

[K.G]

早速の回答ありがとうございました。上手くいきました。詳細はググればなんとか理解できそうです。
pandas の方は確かにコードが短くなり良いですね。これも調べてみます。助かりました。