有没有办法在dandas中生成dtypes作为字典?

时间:2016-12-11 15:36:46

标签: python pandas

输入df.dtypes时,我们有类型列表。 但是,是否有一种简单的方法可以将输出作为

{'col1': np.float32, ...}

或者我是否需要自己编写一个函数?

1 个答案:

答案 0 :(得分:8)

df.dtypes的类型返回对象是pandas.Series。它有一个to_dict方法:

df = pd.DataFrame({'A': [1, 2], 
                   'B': [1., 2.], 
                   'C': ['a', 'b'], 
                   'D': [True, False]})

df
Out: 
   A    B  C      D
0  1  1.0  a   True
1  2  2.0  b  False

df.dtypes
Out: 
A      int64
B    float64
C     object
D       bool
dtype: object

df.dtypes.to_dict()
Out: 
{'A': dtype('int64'),
 'B': dtype('float64'),
 'C': dtype('O'),
 'D': dtype('bool')}

字典中的值来自dtype类。如果您希望将名称作为字符串,则可以使用apply:

df.dtypes.apply(lambda x: x.name).to_dict()
Out: {'A': 'int64', 'B': 'float64', 'C': 'object', 'D': 'bool'}