Python3.x 如何通过Pandas为不同的列集垂直设置列数据

发布于01月19日

我有一个用例，其中我有7个专栏.但前4列在一个DF中，其余3列在第二个DF中.下面是我定义它们的示例代码

import pandas as pd

df1 = pd.DataFrame({
    'col1': [1, 2, 3],
    'col2': ['A', 'B', 'C'],
    'col3': [4.5, 6.7, 8.9],
    'col4': ['X', 'Y', 'Z']
})

df2 = pd.DataFrame({
    'col5': ['apple', 'banana', 'cherry'],
    'col6': [10, 20, 30],
    'col7': [True, False, True]
})
# Concatenate vertically
result_df = pd.concat([df1, df2], axis=0, ignore_index=True)

# Save to Excel
result_df.to_excel('output.xlsx', index=False)

因此，我想以下面的格式插入数据

但是，无论我做什么，我都不能得到这种格式.I实际yields 如下

在我看来，这更像是Pandas 的局限.我有什么办法来达到所要求的结果吗？请提出建议

推荐答案

你不应该使用concat(这将创建一个DataFrame)，而是write the two DataFrames independently.

使用pandas.ExcelWriter处理文件:

# create new excel file
writer = pd.ExcelWriter('output.xlsx')

# write first DataFrame
df1.to_excel(writer, index=False)

# we start to write after the first table
# skipping 1 row (= len(df)+2)
df2.to_excel(writer, startrow=len(df1)+2, index=False)

# close file
writer.close()

generalization

如果您有许多数据帧，您可以概括为:

dfs = [df1, df2, df1, df2]
skip = 1

writer = pd.ExcelWriter('output.xlsx')
start = 0
for df in dfs:
    df.to_excel(writer, startrow=start, index=False)
    start += len(df)+skip+1

writer.close()

NB. this is assuming the columns only have 100 level.

输出:

对于MultiIndex，您还必须保存索引:

dfs = [df1, df2, pd.concat({'extra_level': df1}, axis=1), df1, df2]
skip = 1

writer = pd.ExcelWriter('output.xlsx')
start = 0
for df in dfs:
    df.to_excel(writer, startrow=start, index=True)
    start += len(df)+skip+df.columns.nlevels+(df.columns.nlevels>1)

输出: