pandas获取全部列名,python – 如何在pandas dataframe中获取数字列名

使用select_dtypes和np.number选择所有数字列：

df = pd.DataFrame({'A':list('abcdef'),

'B':[4.5,5,4,5,5,4],

'C':[7.4,8,9,4,2,3],

'D':[1,3,5,7,1,0],

'E':list('aaabbb')})

print (df)

A B C D E

0 a 4.5 7.4 1 a

1 b 5.0 8.0 3 a

2 c 4.0 9.0 5 a

3 d 5.0 4.0 7 b

4 e 5.0 2.0 1 b

5 f 4.0 3.0 0 b

print (df.dtypes)

A object

B float64

C float64

D int64

E object

dtype: object

cols = df.select_dtypes([np.number]).columns

print (cols)

Index(['B', 'C', 'D'], dtype='object')

可以指定float64和int64：

df = pd.DataFrame({'A':list('abcdef'),

'B':[4.5,5,4,5,5,4],

'C':[7,8,9,4,2,3],

'D':[1,3,5,7,1,0],

'E':list('aaabbb')})

df['D'] = df['D'].astype(np.int32)

print (df.dtypes)

A object

B float64

C int64

D int32

E object

dtype: object

cols = df.select_dtypes([np.int64,np.float64]).columns

print (cols)

Index(['B', 'C'], dtype='object')