Python Excelからテキストファイルへの転記について

Question

前提・実現したいこと

ExcelのＡ列の値を１つずつ取り出してテキストファイルのタイトルに入力し、保存したいと考えています。
今のコードですとテキストファイルのタイトルが('A列の値',).txtとなってしまいます。
この()と'',を削除するにはどうしたらいいでしょうか。

該当のソースコード

#! python3
# -*- coding: utf-8 -*-

# excel2txt.py
# 概要:
#   引数に指定したexcelファイルを読み込んで、新規ディレクトリにtxtファイルを作成していきます。
#   指定されたexcelファイルのアクティブシートが対象となります。
#   txtファイルは、A列、B列...の順に作成されていきます。
#   ディレクトリはこのスクリプトを実行した場所に作られます。
#   ディレクトリ名は、「excel2txt_ワークシート名」になります。
#   excelファイルの拡張子は.xlsxのみ対応します。
# 使い方:
#   excel2txt.pyを実行してください。
#   引数にはexcelファイル名を入力してください。
# 実行例:
#   $ ls
#   excel2txt.py    example.xlsx
#   $ python excel2txt.py example.xlsx

import sys
import os
import openpyxl

# コマンドライン引数の取得する
if not len(sys.argv) == 2:
    print('Usage: python excel2txt.py file_name.xlsx')
    print('  One argument must required.')
    print('example: python excel2txt.py example.xlsx')
    sys.exit()
# ファイルパスを変数に格納する
file_path = sys.argv[1]

# ファイルの存在を確認する。
if not os.path.isfile(file_path):
    print(file_path)
    print(file_path + ' is not exist.')
    sys.exit()
# excelファイルとみなす拡張子を設定する
ext = 'xlsx'
# ファイル名の拡張子を確認してxlsxだったら処理を開始する。
if ext in file_path.rsplit('.')[-1]:

    # excelファイルを読み込んでシートのタイトルを取得し、シートの存在を確認する
    try:
        wb = openpyxl.load_workbook(file_path)
        ws = wb.active
        ws_title = ws.title
    except AttributeError:
        print('Sheet dose not exist in "' + file_path + '".')
        sys.exit()

    # ディレクトリを新規作成する
    dir_name = 'excel2txt_' + ws_title
    try:
        os.mkdir(dir_name)
    except FileExistsError:
        print('Directory of ' + dir_name + ' exists.')
        sys.exit()
    # ディレクトリを移動する
    os.chdir(dir_name)

    # A列の値を１つずつテキストファイルのタイトルに入力し保存
    for cell_value in ws.iter_rows(max_col=1, values_only=True):
        file_name = str(cell_value) + '.txt'
        file = open(file_name, mode='w', encoding='utf-8')
        file.close()

else:
    print(file_path + ' is not excel file.')
    sys.exit()

Answer 1

    # シートを列ごとに取得してファイル作成のループ
    for col in range(ws.max_col):
        file_name = ws_title + '_' + str(col) + '.txt'
        file = open(file_name, mode='w', encoding='utf-8')
        for cell in list(ws.cols)[col]:
            file.write(str(cell.value) + '\n')
        file.close()

上記の部分ですが、A列であれば、ws.iter_rows(max_col=1, values_only=True)でループを回し、その値でファイル名を生成すれば良いかと。

for cell_value in ws.iter_rows(max_col=1, values_only=True):
    file_name = str(cell_value) + '.txt'
    file = open(file_name, mode='w', encoding='utf-8')
    file.close()

動作確認はしていません。

Comments

[SuzuAya]

>t_obara様
ご回答ありがとうございます！
ほぼやりたいことが実現できたのですが、ファイル名に('Ａ列の値＇，）.txtというように()と''がついてしまいます。こちらはどう修正すれば消えるでしょうか。

[t_obara]

あ〜tupleで返却されているのでした。以下のページなどを見て値を取得すれば良いかと。
https://note.nkmk.me/python-tuple-list-unpack/

[SuzuAya]

ありがとうございます。無知で恥ずかしい限りですが、値の取得がよく分かりませんでした…。もう少し調べてみます。

Answer 2

pandasを使った方が早そうだったので、コードを丸ごと書き換えたところ、うまくいきました。

import pandas as pd
import numpy as np
import os

def make_traincsv_yolov5(path):
  #train.csvの内容を読み込み、pandasのDataFrameに格納
  df = pd.read_csv(path)

  #重複を排除したリストの作成
  image_id_unique = df.image_id.unique() #重複したimage_idを排除したimage_idリスト作成
  rad_id_unique = df.rad_id.unique() #重複したrad_idを排除したrad_idリスト作成

  #dfに新たな列'sum'を加え、BBoxの情報を格納する
  df.fillna(9999, inplace=True) #まずnullを9999に変えておく(nullのままだと次の行を実行できないため)
  df['sum'] =  df.class_id.astype(str) + ' ' + df.x_min.astype(int).astype(str) + ' ' + df.y_min.astype(int).astype(str) + ' ' + df.x_max.astype(int).astype(str) + ' ' + df.y_max.astype(int).astype(str)
  df.loc[df.class_id==14, 'sum'] = '14 0 0 1 1' #疾患の無い画像の'sum'は、左記のように指定する
  
  #txtファイルの出力先フォルダを作成
  output_dir = './yolov5_txt'
  os.makedirs(output_dir, exist_ok=True)

  #rad_idごとのアノテーション結果をまとめたtxtファイルを作る
  for image in image_id_unique: #for image in image_id_unique[4:5]:
    df2 = df.loc[df.image_id==image]
    for rad in df2.rad_id.unique():
      #新たな画像ファイル名を(元の画像ファイル名)_(rad_id).pngとする
      txt_fname = image + '_' + rad + '.txt' #←新たな画像に対するtxtファイルの名前
      df2.loc[df2.rad_id==rad, 'sum'].to_csv(f'{output_dir}/{txt_fname}', index=None,header=None,sep=",") #←新たな画像に対するtxtファイルを出力

make_traincsv_yolov5("./train.csv/train.csv")