在熊猫中提取dtype为'object'的列的索引

时间:2018-07-27 07:25:28

标签: python pandas numpy

如何从Pandas数据框中的所有列中提取特定dtype的所有列的索引?

1 个答案:

答案 0 :(得分:3)

使用select_dtypes,必要时使用get_indexer

df = pd.DataFrame({'A':list('abcdef'),
                   'B':[4,5,4,5,5,4],
                   'C':[7,8,9,4,2,3],
                   'D':[1,3,5,7,1,0],
                   'E':[5,3,6,9,2,4],
                   'F':list('aaabbb')})

print (df)
   A  B  C  D  E  F
0  a  4  7  1  5  a
1  b  5  8  3  3  a
2  c  4  9  5  6  a
3  d  5  4  7  9  b
4  e  5  2  1  2  b
5  f  4  3  0  4  b

a = df.select_dtypes('object').columns
print (a)
Index(['A', 'F'], dtype='object')

idx = df.columns.get_indexer(df.select_dtypes('object').columns)
print (idx)
[0 5]