Question

所以我正在解析许多文件中包含的数据，循环遍历它们并将某些元素存储在列表中，并使用Python将每个结果列表附加到带有Pandas的数据帧中。

它有效，除了我无法弄清楚如何在追加时保留标题行。它会随着每个追加而消失或重复。

以下代码用作最新代码的示例：

import pandas as pd

for i in range(1,4):
    data = [{'name': 'Company'+str(i), 'city': 'New York'}]

    stuff = []
    for element in data:
        stuff.append(element)

    df = pd.DataFrame(columns=["name",
                               "city"])

    for record in stuff:
        df = df.append(record, ignore_index=True)

    df.to_csv('test.csv', mode='a', header=False, index=False)

使用此代码，输出（csv文件）为：

Company1    New York
Company2    New York
Company3    New York

但我正在寻找输出：

name        city
Company1    New York
Company2    New York
Company3    New York

谢谢！

Answer 1

但您已设置header=False：

df.to_csv('test.csv', mode='a', header=False, index=False)

你应该这样做：

df.to_csv('test.csv', mode='a', header=True, index=False)

您只需要在第一次迭代时将其设置为True，然后在后续迭代中将其设置为False

基本上你只需做以下的事情：

# add this outside your for loop
writeHeader = True

# use the following where you write to csv
if writeHeader is True:
    df.to_csv('test.csv', mode='a', header=True, index=False)
    writeHeader = False
else:
    df.to_csv('test.csv', mode='a', header=False, index=False)

或类似的

所以完整的事情看起来像：

import pandas as pd
writeHeader = True

for i in range(1,4):
    data = [{'name': 'Company'+str(i), 'city': 'New York'}]

    stuff = []
    for element in data:
        stuff.append(element)

    df = pd.DataFrame(columns=["name",
                               "city"])

    for record in stuff:
        df = df.append(record, ignore_index=True)

    if writeHeader is True:    
        df.to_csv('test.csv', mode='a', header=True, index=False)
        writeHeader = False
    else:
        df.to_csv('test.csv', mode='a', header=False, index=False)

保留标题，同时附加到带有Python的Pandas数据帧

1 个答案: