python - 如何在 pandas DataFrame 中存储行名和列索引？

Question

我有一个DataFrame，带有命名行和列索引:

import numpy as np
import pandas as pd

I = pd.Index(["a", "b", "c", "d"], name="rows")
C = pd.Index(["col0", "col1", "col2"], name="cols")
df = pd.DataFrame(data=np.random.rand(4, 3),
                  index=I,
                  columns=C)

我尝试以多种格式(Excel、CSV)存储它，但是当重新读取文件时，名称丢失了(也许我遗漏了一些选项)。 Msgpack 有效，但它被标记为实验性的，所以我现在宁愿避免使用它。我也更愿意避免 pickle。有什么方法(格式和选项)可以存储这两个索引的名称吗？

编辑:我知道如何用 pandas 编写和读取 CSV。问题是保存列索引和行索引的名称。

Answer 1

你可以使用 hdf。

import numpy as np
import pandas as pd
I = pd.Index(["a", "b", "c", "d"], name="rows")
C = pd.Index(["col0", "col1", "col2"], name="columns")
df = pd.DataFrame(data=np.random.rand(4,3), index=I, columns=C)
print(df)

columns      col0      col1      col2
rows                                 
a        0.098497  0.918954  0.642800
b        0.168266  0.678434  0.455059
c        0.434939  0.244027  0.599400
d        0.877356  0.053085  0.182661

df.to_hdf('test.hdf', 'test')
print(pd.read_hdf('test.hdf'))

columns      col0      col1      col2
rows                                 
a        0.098497  0.918954  0.642800
b        0.168266  0.678434  0.455059
c        0.434939  0.244027  0.599400
d        0.877356  0.053085  0.182661