get list of pandas dataframe columns based on data type

前端 未结 12 1401
后悔当初
后悔当初 2020-12-07 07:37

If I have a dataframe with the following columns:

1. NAME                                     object
2. On_Time                                      object
         


        
12条回答
  •  情话喂你
    2020-12-07 07:53

    As of pandas v0.14.1, you can utilize select_dtypes() to select columns by dtype

    In [2]: df = pd.DataFrame({'NAME': list('abcdef'),
        'On_Time': [True, False] * 3,
        'On_Budget': [False, True] * 3})
    
    In [3]: df.select_dtypes(include=['bool'])
    Out[3]:
      On_Budget On_Time
    0     False    True
    1      True   False
    2     False    True
    3      True   False
    4     False    True
    5      True   False
    
    In [4]: mylist = list(df.select_dtypes(include=['bool']).columns)
    
    In [5]: mylist
    Out[5]: ['On_Budget', 'On_Time']
    

提交回复
热议问题