【Pandas】データフレームから特定範囲の行・列を抽出

この記事では、PythonモジュールPandasで、データフレームから特定範囲の行・列を抽出する方法をソースコード付きで紹介します。

データフレームから特定範囲の行・列抽出

PythonモジュールPandasでは、locメソッドで特定の行・列(インデックス・カラム)の範囲だけを取り出せます。

書式

df.loc[[ インデックス1, インデックス2, ...],  [カラム1, カラム2, ...]]

返り値：指定した範囲内のデータフレーム

ソースコード

サンプルプログラムのソースコードです。

# -*- coding: utf-8 -*-
import pandas as pd

def main():
    # リスト生成
    name = ['西住みほ', '秋山優花里', '武部沙織']
    height = [158, 157, 157]
    position = ['車長', '装填手', '通信手']

    # リスト→データフレーム
    df = pd.DataFrame({'名前' : name, '身長' : height}, index = position)

    # データフレーム表示
    print(df)

    '''
        　　　　名前   身長
    車長    西住みほ  158
    装填手  秋山優花里  157
    通信手   武部沙織  157
    '''
    indexes = ['車長', '装填手']
    # ラベル（列）を抽出
    df2 = df.loc[indexes, '名前']
    # 抽出したデータフレームを表示
    print(df2)
    '''
    車長      西住みほ
    装填手    秋山優花里
    Name: 名前, dtype: object
    '''

if __name__ == "__main__":
    main()