python - 添加文件名作为 CSV 文件的最后一列

Question

我有一个 Python 脚本，它修改 CSV 文件以将文件名添加为最后一列:

import sys
import glob

for filename in glob.glob(sys.argv[1]):
    file = open(filename)
    data = [line.rstrip() + "," + filename for line in file]
    file.close()

    file = open(filename, "w")
    file.write("\n".join(data))
    file.close()

不幸的是，它还将文件名添加到文件的标题(第一)行。我想将字符串“ID”添加到标题中。谁能建议我如何做到这一点？

Answer 1

看看官方csv module .