保存后数据帧类型丢失

Question

我有我更改为类别类型的数据框。但是在保存和加载后类型返回到 float64

data = pd.DataFrame(barray, columns=['line 1', 'line 2', 'line 3', 'line 4', 'piece to move', 'place to occupy'])
    data['possibility of win'] = y
    for col in ['line 1', 'line 2', 'line 3', 'line 4', 'piece to move', 'place to occupy']:
        data[col] = data[col].astype('category')

    data['line 1'].dtype

输出： CategoricalDtype(categories=['2000', '2001', '2002', '2010', '2011', '2012', '2020', '2021', '2100', '2101', '2102', '2110', '2112', '2120', '2121'、'2200'、'2201'、'2202'、'2210'、'2211'、'2212'、 '2220', '2221'], 有序=假）

data.to_csv('data.csv')
data = pd.read_csv("data.csv")
data['line 1'].dtype

输出： dtype('int64')

Answer 1

这是意料之中的，因为在 csv 中所有数据都像文本一样保存。

解决方法是在read_csv中添加参数dtype：

#all columns set to categoricals
data = pd.read_csv("data.csv", dtype='category')

如果只想指定某些列使用字典：

cols = ['line 1', 'line 2', 'line 3', 'line 4']
d = dict.fromkeys(cols, 'category')
data = pd.read_csv("data.csv", dtype=d)