繁体   English   中英

pandas - 将附加列添加到现有 csv 文件

[英]pandas - add additional column to an existing csv file

我有 2 个文件需要填充相同的 csv 文件

为了给出上下文,目前,我有这个代码以所需的方式打印 CSV code-1.py

当前现有添加:

array_all = {'Hand': alldata, 'Pose': face_position}
            array_all = {k: pd.Series(v) for k, v in array_all.items()}
            df = pd.DataFrame(array_all)

            df.to_csv('test.csv',
                        mode='w',
                        header=True,
                        index=False)
姿势
靠背座椅
靠背座椅

现在我有 code-2.py

要添加的列:

   df = pd.DataFrame(results)

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

我想要做的是在右侧添加一列 Desired Output:

姿势 眼睛
靠背座椅 剩下
靠背座椅 正确的

但是,目前我得到了这个,

实际 Output:

姿势
靠背座椅
靠背座椅
0
正确的
剩下

基本上,它附加到 CSV 的第一列此外,可以假设 code-2.py 将在 code-1.py 之后立即运行

感谢您对此的任何想法,谢谢!

您不能将 append 列添加到 csv 文件而不完全加载它(但是,您可以 append 行)。 使用pd.concat

pd.concat([pd.read_csv('test.csv'), df], axis=1) \
  .to_csv('test.csv', header=True, index=False)
# test.csv before
Hand,Pose
No,Seating Back
No,Seating Back


# test.csv after
Hand,Pose,Eye
No,Seating Back,Left
No,Seating Back,Right

参考@Corralien的回答

我还遇到了一个问题,即每个额外的 append 都会记录在一个新列中,如此处所示https://i.stack.imgur.com/08oTe.png

为了缓解这种情况,我修改了给定的答案并创建了一个额外的 csv 并使用它覆盖了 csv ,如下所示:

     pd.concat([pd.read_csv('test2.csv'), df], axis=1) \
            .to_csv('test.csv', header=True, index=False)
df = (pd.read_csv("test.csv")).assign(Eyes=results)
df.to_csv('test.csv', header=True, index=False)

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM