Python を使用して CSV ファイルから特定の列を抽出する方法

Python csv モジュールを使用した CSV 列のトリミング

多くの場合、カンマ区切り値 (CSV) ファイルを解析し、データの特定の列を抽出します。 Python の csv モジュールを使用すると、このタスクを簡単に実行できます。

列の抽出

特定の列を抽出するには、include_cols 変数を利用します。これは、列インデックスを指定します。含まれています。各列には 0 から始まる番号が付けられ、最初の列は 0 になります。たとえば、「ID」、「Name」、「Zip」、および「Phone」列をキャプチャするには、include_cols = [1, 2, 6] と記述します。 , 7].

反復と印刷

行を反復処理して指定された列を抽出するには、for ループ内でリスト内包表記を使用します。各行について、指定された列の値が content 変数に設定されます。抽出されたデータを表示するには、ループ内で print(content) を使用します。

included_cols = [1, 2, 6, 7]

for row in reader:
    content = list(row[i] for i in included_cols)
    print(content)

Pandas による最適化

より効率的に CSV を処理するには、pandas モジュールの使用を検討してください。 。 Pandas は、CSV ファイルの読み取りと操作のための便利で強力なツールを提供します。 CSV を読み取り、特定の列を変数に保存するには、次を使用します:

import pandas as pd
df = pd.read_csv(csv_file)
saved_column = df.column_name

以上がPython を使用して CSV ファイルから特定の列を抽出する方法の詳細内容です。詳細については、PHP 中国語 Web サイトの他の関連記事を参照してください。