PythonでCSVファイルから特定の列を抽出する方法（Pandas活用）

PythonでCSVファイルから特定の列だけを抽出してリストとして扱いたい場合、Pandasのread_csv()メソッドを使うと簡単に実現できます。usecols引数に列名のリストを指定することで、必要な列のみを効率的に読み込むことが可能です。

手順

• 抽出したい列名をリストとして作成します。

• read_csv()メソッドを使って、CSVファイルをデータフレーム（DataFrame）として読み込みます。

• 抽出したデータを表示します。

• plot()メソッドを使ってデータフレームをグラフ化します。

• 図を表示するためにshow()メソッドを呼び出します。

コード例

import pandas as pd
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
columns = ["Name", "Marks"]
df = pd.read_csv("input.csv", usecols=columns)
print("Contents in csv file:\n", df)
plt.plot(df.Name, df.Marks)
plt.show()

サンプルデータ（input.csv）

この例で使用するCSVファイルには、以下のようなデータが含まれています。

実行結果

コードを実行すると、CSVファイルから「Name」と「Marks」の2列だけが抽出され、コンソールにはデータフレームの内容が出力されます。同時に、matplotlibによって各名前の点数を示す折れ線グラフが表示されます。このようにusecolsを活用すれば、大きなCSVファイルからも必要なデータだけを素早く取り出せます。