CSVファイルから特定データの検索をPythonで実現する方法

CSVファイルの特定のデータを探すには、Pythonのcsvモジュールを使用してCSVファイルを読み込み、ファイルの各行をループで反復処理し、行ごとにターゲットデータを検索します。以下にそのサンプルコードを示します。

import csv

def find_data(csv_file, target_data):
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        for row in reader:
            if target_data in row:
                return row
    return None

# 示例用法
csv_file = 'data.csv'
target_data = 'John'
result = find_data(csv_file, target_data)
if result:
    print("找到数据：", result)
else:
    print("未找到数据")

上の例では、CSVファイルのパスと目的のデータを2つの引数にとるfind_dataという関数を定義しています。関数はwith openステートメントを使用してCSVファイルを開き、csv.readerを使用してリーダーオブジェクトを作成します。次に、各行をループで繰り返し処理し、目的のデータが現在の行にある場合はその行のデータを返します。ループ終了後も目的のデータが見つからない場合、Noneを返します。

サンプルの使い方ではCSVファイルのパスと検索したいデータを渡し、返ってきた結果がNoneじゃないかを確認し、そうでなければ見つかったデータをプリントしています。