相关疑难解决方法(0)

如果"filename"不存在,我想使用pd.write_csv写"filename"(带标题),否则如果存在则附加到"filename".如果我只是使用命令:

     df.to_csv('filename.csv',mode = 'a',header ='column_names')

写入或追加成功,但似乎每次发生追加时都会写入标题.

如果文件不存在,我怎么才能添加标题,如果文件存在则不添加标题？

所以我正在解析许多文件中包含的数据,循环遍历它们并将某些元素存储在列表中,并使用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

谢谢!