[英]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.