我有一个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()
Run Code Online (Sandbox Code Playgroud)
不幸的是,它还将文件名添加到文件的标题(第一行).我希望将字符串"ID"添加到标题中.谁能建议我怎么做?