Selecting Pandas Columns by dtype

后端 未结 9 2067
爱一瞬间的悲伤
爱一瞬间的悲伤 2020-11-29 01:22

I was wondering if there is an elegant and shorthand way in Pandas DataFrames to select columns by data type (dtype). i.e. Select only int64 columns from a DataFrame.

<
9条回答
  •  长情又很酷
    2020-11-29 01:48

    Since 0.14.1 there's a select_dtypes method so you can do this more elegantly/generally.

    In [11]: df = pd.DataFrame([[1, 2.2, 'three']], columns=['A', 'B', 'C'])
    
    In [12]: df.select_dtypes(include=['int'])
    Out[12]:
       A
    0  1
    

    To select all numeric types use the numpy dtype numpy.number

    In [13]: df.select_dtypes(include=[np.number])
    Out[13]:
       A    B
    0  1  2.2
    
    In [14]: df.select_dtypes(exclude=[object])
    Out[14]:
       A    B
    0  1  2.2
    

提交回复
热议问题