pandas で任意の列を抽出する
Python
Published: 2019-06-29

やったこと

pandas で任意の列を抽出してみます。

確認環境

$ ipython --version
6.1.0
$ jupyter --version
4.3.0
$ python --version
Python 3.6.2 :: Anaconda custom (64-bit)
import pandas
print(pd.__version__)
0.20.3

調査

アイリスデータの1 ~ 2列目を取得します

import pandas
import pandas as pd
from sklearn import datasets
iris = datasets.load_iris()
train = iris.data

df = pd.DataFrame(train)
df[[0, 1]]

また、下記でも同じものが取得できます。

df.loc[:, [0, 1]]

参考