pandasよく使う機能メモ

★DataFrame操作

import pandas as pd

 

dict_test = [

    {

        'time': '2020',

        'id': 1,

        'name': 's1'

    },

    {

        'time': '2021',

        'id': 2,

        'name': 's2'

    }

]

• DataFrame作成

df_test = pd.DataFrame(dict_test)

 

　time id name
0 2021 1 s1
1 2020 2 s2

• 列をindexに設定

df_test = df_test.set_index('time')

 

　 id name
time
2021 1 s1
2020 2 s2

• indexでsort

df_test = df_test.sort_index(ascending=True)

 

　id name
time
2020 2 s2
2021 1 s1

•  列名の条件で列抽出 

df_name = df_test.loc[:, df_test.columns.str.endswith('name')]

 

　　name
time
2020 s2
2021 s1