python - 在 Pandas 中创建指定列类型的空数据框

Question

我正在尝试创建一个带有索引的空数据框并指定列类型。我这样做的方式如下:

df = pd.DataFrame(index=['pbp'],
                  columns=['contract',
                           'state_and_county_code',
                           'state',
                           'county',
                           'starting_membership',
                           'starting_raw_raf',
                           'enrollment_trend',
                           'projected_membership',
                           'projected_raf'],
                  dtype=['str', 'str', 'str', 'str',
                         'int', 'float', 'float',
                         'int', 'float'])

但是，我收到以下错误，

TypeError: data type not understood

这是什么意思？

Answer 1

您可以使用以下内容:

df = pd.DataFrame({'a': pd.Series(dtype='int'),
                   'b': pd.Series(dtype='str'),
                   'c': pd.Series(dtype='float')})

或更抽象地说:

df = pd.DataFrame({c: pd.Series(dtype=t) for c, t in {'a': 'int', 'b': 'str', 'c': 'float'}.items()})

那么如果你调用 df 你有:

>>> df 
Empty DataFrame 
Columns: [a, b, c]
Index: []

如果你检查它的类型:

>>> df.dtypes
a      int32
b     object
c    float64
dtype: object