将层次化索引的DataFrame保存为csv文件再读取时会丢失索引格式,通过设置header和index_col参数保留原索引信息。
一.设置多重索引的DataFrame并保存
from pandas import DataFrame
import pandas as pd
import numpy as np
index = [['x']*2+['y'],['n_1','n_2','n_1']]
columns = [['a']*3+['b']*2,['m_1','m_2','m_3','m_1','m_2']]
df1 = DataFrame(np.arange(15).reshape((3,5)), index=index, columns=columns)
df1.to_csv('target.csv',encoding='utf-8')
df1
二.普通读取的索引格式缺失
df2 = pd.read_csv('target.csv',encoding='utf-8')
df2
三.指定header和index_col参数,第1、2行为columns,第1,2列为index
df2 = pd.read_csv('target.csv',encoding='utf-8',header=[0,1],index_col=[0,1])
df2