0

我有以下代码

    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()

现在我想对名为“CSV”的文件夹中包含的所有文件执行它。我正在研究 UBUNTU 12.04 和 linux 的新手。

所有的帮助将不胜感激。谢谢。

4

2 回答 2

0

我不知道您要达到什么目的，但是如果您想遍历文件夹中的大量文件，可以使用 linuxfind命令。

在您的情况下的一个例子是：

cd /path/to/folder/CSV
find . -name *.csv -exec your_python_script.py {} \;

这将找到所有.csv文件并在每个文件上运行您的 python 脚本。

确保您#!/bin/env python在文件顶部有并运行sudo chmod +x your_python_script.py以使其可执行。

如果您没有.csv扩展名，则可以执行以下操作：

cd /path/to/folder/CSV
find . -exec your_python_script.py {} \;

尽管被警告 - 这将在目录中的每个文件上运行- CSV 或其他文件。

另外，如果我正确理解了您的脚本（它只是将文件名附加到最后一列），请尝试以下操作：

#!/bin/env python

import sys
import csv

filename=sys.argv[1]
output_filename='new-'+filename
output_rows = []
with open(filename, 'r') as inputfile:
    with open(output_filename) as outputfile:
        r = csv.reader(inputfile)
        w = csv.writer(outputfile, lineterminator='\n')
        for input_row in r:
            output_row=input_row.append(filename)
            output_rows.append(output_row)

        w.writerows(output_rows)

将其保存test_python_script.py在您的CSV文件夹中。然后运行：

cd /path/to/folder/CSV
chmod +x test_python_script.py
find . -exec test_python_script.py {} \;
于 2013-05-30T08:46:38.437 回答
0

你也可以简单地让 shell 为你做 globbing：

python myscript.py path/to/files/*.CSV

然后在你的python脚本中：

[...]
for filename in sys.argv[1:]:
    # do stuff...

但是，这在 Windows 中不起作用，因为您只会在 sys.argv 中收到一个参数“path/to/files/*.CSV”。

于 2013-05-30T09:25:44.413 回答