CSV モジュールを使用して CSV ファイルの特定の列からデータを抽出するにはどうすればよいですか?

CSV モジュールを使用して CSV ファイルの特定の列からデータを抽出する

CSV ファイルを操作する場合、多くの場合、データを抽出する必要があります。特定の列からのみ。 CSV モジュールを使用してこれを実現するには、ライブラリがデータ取得をどのように処理するかを理解することが重要です。

提供されたコード サンプルでは、​​row[i] 構文を通じて特定の列にアクセスしようとしています。ここで、i は列番号。ただし、アプローチが間違っているため、結果は期待どおりではありません。

正しい方法は、for ループ内に print ステートメントを含めることです。次の修正されたコードは、これを示しています。

import csv

included_cols = [1, 2, 6, 7]
with open("csv_file.csv") as csvfile:
    reader = csv.reader(csvfile, delimiter=",")
    for row in reader:
        content = list(row[i] for i in included_cols)
        print(content)  # Include the print statement within the loop to display each row

修正されたコードは、各行の必要な列の値を出力します。

ただし、 CSV ファイルの処理を簡素化する pandas ライブラリ。 pandas を使用すると、CSV ファイルを簡単に読み取り、特定の列を変数に抽出できます。

import pandas as pd

df = pd.read_csv("csv_file.csv")
names = df["Name"]  # Use df['column_name'] to extract a specific column into a variable

pandas ライブラリは、構造化データを操作するための多用途のツール セットを提供するため、CSV ファイルを処理する場合に推奨されます。効果的に。

以上がCSV モジュールを使用して CSV ファイルの特定の列からデータを抽出するにはどうすればよいですか?の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。