Python:从csv文件中读取Pandas Dataframe,将过滤后的输出作为csv过滤到另一个文件

时间:2017-11-23 16:02:55

标签: python pandas csv

我有一堆N csv文件,我想读入内存,过滤某些特征,并通过附加M csv文件写入 - 格式与输入文件相同。

如果我使用

df = pd.read_csv
...
for index, line in df.iterrows():
    with open(fullOutputFileName, 'a') as f:
        line.to_csv(fullOutputFileName, index.None)

我没有一条输出线

attr1, attr2, attr3

对于一个输入行,但类似于:

attr1
attr2
attr3

(因为这可能是熊猫系列写入文件系统的方式)。

还有其他方法可以执行此任务(可能没有Pandas)吗?

1 个答案:

答案 0 :(得分:0)

不要尝试单独编写每一行,数据框有to_csv方法。

df = pd.read_csv('input.csv')
# some filtering logic, for example:
filtered_df = df[df['col a'] == 2]
filtered_df.to_csv('output.csv')