输入df.dtypes
时,我们有类型列表。
但是,是否有一种简单的方法可以将输出作为
{'col1': np.float32, ...}
或者我是否需要自己编写一个函数?
答案 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'}