【業務自動化】チャプター2-3：置換するファイル形式を選択式にしよう

一つ前のページでは 表記ゆれ置換ツール を作成しました。

今回は 置換対象ファイルをCSVにも拡張 しましょう。

前回の記事では、Excelで管理している「表記ゆれリスト」を使い、テキストファイル内の用語を一括で置換するツールを作成しました。

しかし、実務では扱うデータが必ずしもテキストファイルとは限りません。

CSVファイル（例えば顧客リストや取引履歴など）を対象に同様の置換を行いたい場面もよくあります。

そこで今回の記事では、ユーザーが処理対象のファイル形式（txt / csv）を選べるようにし、CSVにも対応したツールに改良する方法を学びます。

新しく導入する ライブラリはPyInputPlus です。これは入力値をチェックしながら受け付ける外部モジュールで、ユーザーに誤った入力をさせない仕組みを簡単に追加できます。

記事を読み終える頃には、以下のようなスキルが身についているでしょう。

• 既存のコードを拡張して機能を追加する手順
• PyInputPlusを使った選択式メニューの作成
• CSVファイルをPythonで安全に読み書きする方法

<<前のページ

業務自動化の記事一覧

次のページ>>

新規モジュールの基本｜PyInputPlusとcsvモジュール

サンプルスクリプトを紹介する前に、今回の記事で新たに登場する２つのモジュール、pyInputPlusとcsvを紹介します。

あわせて読みたい

openpyxl使える関数・メソッド・クラス一覧【Excel自動化の基本リファレンス】

pyInputPlusモジュールの紹介

PyInputPlusモジュール は、Python の外部ライブラリで、input()関数を拡張して入力の検証や制御を簡単に行えるようにしたものです。

ユーザーからの入力に対して、型チェック・範囲指定・再入力制御などを組み込みでサポートします。

例えば、pyInputPlusモジュールに含まれる inputInt()関数 を使って入力をすると、入力値が整数でなかった場合は自動的にエラーを出して再入力を促します。

つまり、ユーザーが文字列や浮動小数点数などを入れても、勝手に int()変換エラーで落ちることはなく、エラーメッセージを表示してもう一度入力待ちに入ります。

PyInputPlusのインストール手順とサンプルコード

コマンドプロンプトに以下のコードを打ち込んで、pyinputplusをインストールしましょう。

pip install pyinputplus

インストールができたら、以下のコードをコピーして実行し、いろいろな入力を試してみて下さい。

通常の input()関数より安全（型変換エラーや不正値を防げる）であり、再入力処理のロジックを自分で書かなくて済む点を体感しましょう。

import pyinputplus

# 整数入力（1～10の範囲）
age = pyinputplus.inputInt("年齢を入力してください (1-10): ", min=1, max=10)
print(f"あなたの年齢: {age}")

# 選択肢から入力
fruit = pyinputplus.inputChoice(['apple', 'banana', 'orange'])
print(f"選ばれたフルーツ: {fruit}")

# Yes/No 入力
proceed = pyinputplus.inputYesNo("続行しますか？ (yes/no): ")
if proceed == "yes":
    print("続行します！")
else:
    print("終了します。")

PyInputPlus 関数一覧表

PyInputPlusに含まれる主要な関数を、表形式で整理します。

自動化ツールを作る際には非常に便利ですので、是非使用していきましょう。

csvモジュールの基本｜reader/writerでセル単位の置換

csvモジュール は、Pythonに標準で搭載されている（追加インストール不要）モジュールで、CSVファイル の読み書きを簡単に行うためのツールです。

主な特徴は以下の通りです。

csv.readerの使い方｜UTF-8 / BOMなど文字コードの扱い

例えば、単純にCSVファイルを読み込むコードは以下のようになります。

import csv

# CSVファイルを読み込む
with open("data.csv", newline="", encoding="utf-8-sig") as f:	# UTF-8（BOM付き）でファイルを開く
    reader = csv.reader(f)  # リーダーオブジェクトを作成
    rows = [row for row in reader]  # 行をリストに格納

print(rows)

このコードはdata.csvを読み込み、その中身のデータを変数rowsに格納して出力しています。

１．with open(“data.csv”, newline=””, encoding=”utf-8-sig”) as f:

この行によって、CSVファイルの中身を読み取れる準備が整います。

２．reader = csv.reader(f)

３．rows = [row for row in reader]

結果として rows は「リストの中にリストが入っている形（二次元リスト）」になります。

csv.writerの使い方｜安全な上書き防止と新規ファイル出力

単純にCSVファイルに書き込むコードは以下のようになります。

import csv

# 書き込みたいデータ（2次元リスト）
rows = [
    ["名前", "年齢", "出身地"],
    ["太郎", "20", "東京"],
    ["花子", "25", "大阪"]
]

# CSVに書き込み
with open("output.csv", "w", newline="", encoding="utf-8-sig") as f:
    writer = csv.writer(f)   # writerオブジェクトを作成
    writer.writerows(rows)   # まとめて書き込み

このコードはoutput.csvを読み込み、変数rowsのデータを書き込んでいます。

１．with open(“output.csv”, “w”, newline=””, encoding=”utf-8-sig”) as f:

２．writer = csv.writer(f)

３．writer.writerows(rows)

置換ツールをCSV対応に改良｜選択して一括置換

まずは、これまでテキストファイルのみを対象にしていた表記ゆれ一斉置換ツールを、CSVファイルにも対応できるように修正していきます。

具体的には、プログラムの冒頭でユーザーに「txt」か「csv」を選ばせ、その入力値に応じて処理を分岐させます。