【Pandas】データフレームから特定の列(ラベル)を抽出

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

データフレームから列抽出

PythonモジュールPandasでは、主に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
    '''

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

if __name__ == "__main__":
    main()