使用MultiIndexing修复导出到Excel

时间:2015-05-20 23:53:06

标签: python excel python-2.7 pandas

所以到目前为止这是我的问题。我的Pandas版本是0.15.2,我正在使用Python 2.7。我正在尝试将具有MultiIndexed列的数据框导出为ex​​cel而不使用垂直索引。这完全概括了我的问题:

# imports
>>> import pandas as pd
>>> import numpy as np
# setting columns
>>> level_one = ['a']*3 + ['b']*3 + ['c']*3
>>> level_two = ['1', '2', '3']*3
>>> columns = [np.array(level_one), np.array(level_two)]
# making the dataframe
>>> df = pd.DataFrame(index=range(10), columns=columns)
>>> series = pd.Series({k:k**2 for k in range(10)})
>>> df[first_col] = series
>>> first_col = df.columns.tolist()[0]
>>> df
     a              b              c          
     1    2    3    1    2    3    1    2    3
 0   0  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 1   1  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 2   4  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 3   9  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 4  16  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 5  25  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 6  36  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 7  49  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 8  64  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN
 9  81  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN

所以,到目前为止一切正常。现在,当我导出到csv时,两种方法都在数据完整性方面起作用(相对于标题没有数据移位,尽管它确实重复了level_one中的每个元素)。

 >>> import os
 >>> path = os.path.join(os.path.expanduser('~'), 'test.csv')
 >>> df.to_csv(path)
 >>> df.to_csv(path, index=False)

但是,对于ExcelWriter,它执行以下操作:

 >>> import os
 >>> path = os.path.join(os.path.expanduser('~'), 'test.xlsx')
 >>> writer = pd.ExcelWriter(path)
 >>> df.to_excel(writer, 'Sheet 1')
 >>> df.to_excel(writer, 'Sheet 2', index=False)
 >>> writer.save()

With Index

Index False

正如您所看到的,它消除了索引但不在列标题中,从而导致数据移位。有修复或解决方法吗?在我的实际数据集中,实际的“索引”具有重复的元素和间隔符,因此它不适合作为索引使用。

1 个答案:

答案 0 :(得分:0)

所以我在文档中找不到任何解决这个问题的方法,但我想出了一个临时解决方法。

import itertools

def pseudo_header(df):
    '''Create a pseudo-header for the dataframe due to indexing
    issues.
    '''

    # grab grouped columns
    columns = df.columns
    grouped = itertools.groupby(columns, key=lambda x: x[0])
    grouped = [tuple(v) for k, v in grouped]
    # grab col/index counters
    index = DF.get_last_index(df)
    counter = 0
    # set rows
    df.loc[index] = pd.Series(index=columns)
    df.loc[index+1] = pd.Series(index=columns)
    for group in grouped:
        for idx, values in enumerate(group):
            # grab indexing
            column = columns[counter]
            if idx == 0:
            df.loc[index, column] = values[0]
            df.loc[index+1, column] = values[1]
            counter += 1

然后,我导出索引和标题关闭。它有效,没有标题格式,但这不是一个糟糕的修复。

目前,只适用于2级标题,我可以概括,但根据我的需要,现在没有必要。