主页 / ubuntu / 问题 / 646876
Accepted
Lynob
Asked: 2015-07-11 09:09:43 +0800 CST

将文件名从 bash 传递给 python

  • 772

可以说我在一个文件夹中有 1000 个具有随机名称的 csv 文件test。我有一个 python 脚本script.py。我想写一个 bash 脚本来传递文件夹中每个文件的名称script.py然后运行它，因此 bash 脚本应该运行script.py1000 次，每次都使用不同的文件名，例如，第一次运行：

with open('records.csv','r') as in_file

下次运行

with open('vertigo.csv','r') as in_file

我知道我可以在 python 中完成它而无需像这样编写 bash 脚本

import glob
for filename in glob.glob('*.csv'):
   # script here

但我正在做一些测试，因此我想在 bash 中进行

bash

5 个回答

  1. Best Answer

    使用sys.argv[1]以便从参数中获取文件名：

    import sys

with open(sys.argv[1],'r') as in_file

    然后，您可以使用各种方法将文件名作为参数传递。例如，与find：

    find test/ -type f -name '*.csv' -exec /path/to/script.py {} \;
    • 6

  2. 使用的替代方法sys.argv是使用argparse. 如果您需要更多的命令行参数解析，这非常有帮助。

    #!/usr/bin/env python2

import argparse

def main(files):
    for f in files:
        print "Would do something with", f

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('file', nargs='+', help='path to the file')
    args_namespace = parser.parse_args()
    args = vars(args_namespace)['file']
    main(args)

    通过 Bash 扩展我们的文件参数列表来运行它：

    $ ./script.py *.csv
Would do something with file1.csv
Would do something with file2.csv
Would do something with file3.csv

    并“免费”提供内置帮助：

    $ ./script.py --help
usage: script.py [-h] file [file ...]

positional arguments:
  file        path to the file

optional arguments:
  -h, --help  show this help message and exit
    • 3

  3. 你必须在 python 脚本中检索文件名才能工作：

    #!/bin/bash
for f in *.csv; do
    python script.py "$f"
done

    在一行中：

    for f in *.csv; do python script.py "$f"; done
    • 2

  4. 扩展 muru 的回答：

    为什么不在你的 python 脚本中完成这一切？将脚本对文件的操作转换为函数，然后以目录 作为参数运行脚本：

    #!/usr/bin/env python3
import os

#--- your original script, transformed into a function
def some_function(file):
    print(file)
#---

# The directory with your .csv files
dr = sys.argv[1]

for f in os.listdir(dr):
    file = dr+"/"+f
    some_function(file)

    或者更短一点：

    for f in os.listdir(dr):
    some_function(dr+"/"+f)

    或者，如果该目录还可能包含其他文件：

    for f in [f for f in os.listdir(dr) if f.endswith(".csv")]:
    some_function(dr+"/"+f)

    然后只需将目录作为参数运行它：

    python3 <script> <directory>
    • 2

  5. 使用sys.argvas 列表，提供所有单独的文件作为参数，并让您的 shell（例如 Bash）为您展开它。

    script.py好像：

    #!/usr/bin/env python2

import sys

def main(files):
    for f in files:
        print "Would do something with", f

if __name__ == "__main__":
    files = sys.argv[1:]: # slices off the first argument (executable itself)
    main(files)

    像这样的文件树：

    .
├── file1.csv
├── file2.csv
├── file3.csv
└── script.py

0 directories, 4 files

    然后使其可执行：

    chmod +x script.py

    使用 shell 扩展的参数运行它：

    ./script.py *.csv

    输出：

    Would do something with file1.csv
Would do something with file2.csv
Would do something with file3.csv
    • 2

相关问题