所以到目前为止这是我的问题。我的Pandas版本是0.15.2,我正在使用Python 2.7。我正在尝试将具有MultiIndexed列的数据框导出为excel而不使用垂直索引。这完全概括了我的问题:
# 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()
正如您所看到的,它消除了索引但不在列标题中,从而导致数据移位。有修复或解决方法吗?在我的实际数据集中,实际的“索引”具有重复的元素和间隔符,因此它不适合作为索引使用。
答案 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级标题,我可以概括,但根据我的需要,现在没有必要。