pd.read_csvでカラム指定したい

2024年6月8日

pd.read_csvを使用してCSVファイルを読み込む際に、特定のカラムだけを指定して読み込む方法があります。以下に例を示します。

python
import pandas as pd

# CSVファイルを読み込む際に特定のカラムだけを指定する
# usecolsには指定したいカラムのインデックスやカラム名をリストで渡す
df = pd.read_csv('data.csv', usecols=['column_name1', 'column_name2'])

# カラムのインデックスを指定する場合
# df = pd.read_csv('data.csv', usecols=[0, 1])  # 0と1番目のカラムを読み込む

# 読み込んだデータを表示
print(df)

usecols引数には、読み込みたいカラムのインデックス番号やカラム名をリストとして渡します。これにより、指定したカラムだけが読み込まれます。