python - 带有 pandas 的 CSV 副本

Question

我知道这个主题已被广泛处理，但我无法得到我想要的，对于可能是新手问题感到抱歉。所以问题是我有一个像这样的 CSV:

Date,"Tmax","Tmin","Tmedia","Rachas","Vmax","LT","L1","L2","L3","L4"
23 nov 2018,"14.0 (15:30)","7.3 (23:59)","10.7","12 (14:50)","5 (14:50)","2.0","1.6","0.4","0.0","0.0"

我每天都会收到一个像这样的新 CSV，其中包含多行，但我仅对标题后的第一行感兴趣。我想要做的就是每天将第一行迭代地复制到新的 CSV 中，因此在本周末，该 CSV 应该有七行。此外，我想检查该日期是否已在每日文件中。问题是我没有正确获取新的 CSV，这是我的尝试:

import pandas as pd

df = pd.read_csv('file.csv', skiprows=4, header=None)
writer=df[df.index.isin([0])].to_csv('output.csv',header=None)

这段代码的问题是它每次都会覆盖文件output.csv。然后我考虑将其更改为:

writer=df[df.index.isin([0])]
pd.read_csv('output.csv').append(writer).to_csv('output.csv',header=None)

现在的问题是它确实需要该文件先前存在；即使如此，信息也没有正确复制到新文件中。我认为它一定比这更简单，但我被困住了。感谢您的帮助。

Answer 1

如果您只需要标题后的第一行，请读取标题并仅使用nrows=1。然后在附加模式下使用 open 将单行数据帧写入 csv 文件的末尾。 header=False 参数可以很好地处理写入时排除 header 的问题。

df = pd.read_csv('file.csv', nrows=1)

with open('output.csv', 'a') as fout:
    df.to_csv(fout, header=False)

我省略了 skiprows=4，因为不清楚它与您的输入数据有何关系。