使用多行索引和多列索引讀取和寫入Pandas DataFrame

Question

我有一個帶有多行索引和多列索引的熊貓數據幀。

行索引名稱：（User_ID，User_Name）。列索引名稱：（User_Type，Sample_ID）。

數據框看起來像這樣。

In: user_frame.head(2) 

Out: 

     Sample_Type TV TV Radio TV 
     Sample_ID 1 2 3  4 
User_ID User_Name 
1000001 Jim   0.1 0.3 0.5 0.9 
1000001 Julie  0.2 0.9 0.1 0.5 

我想寫這個到磁盤。

user_frame.to_csv(fi_path,{write_args}) 

然後用多行和多列索引讀回來。

user_frame_new=pd.read_csv(fi_path,{read_args}) 

我還沒有找到保留多行和多列索引的正確write_args/read_args。

這樣做的正確方法是什麼？

由於

Answer 1

下面是一個例子：

idx = pd.MultiIndex.from_product([["A", "B"], [1, 2, 3]]) 
col = pd.MultiIndex.from_product([["X", "Y"], ["aa", "bb"]]) 
df = pd.DataFrame(np.random.randint(0, 100, size=(6, 4)), index=idx, columns=col) 
df.index.names = "name1", "name2" 
df.columns.names = "NAME1", "NAME2" 

df.to_csv("tmp.csv") 
df2 = pd.read_csv("tmp.csv", header=[0, 1], index_col=[0, 1])