在python中执行linux time命令

Question

我想计算在 python 脚本中执行 c 程序所需的时间。我为此使用了 os.system() 函数。

os.system("{ time ./test.out < inp;} > out 2> exe_time")

• test.out 是我的 c 可执行文件
• inp 包含 c 的输入
• out存储c程序的输出
• exe_time 存储程序的执行时间。

我在 exe_time 中得到的结果是这样的

0.00user 0.00system 0:00.00elapsed ?%CPU (0avgtext+0avgdata 1416maxresident)k 0inputs+8outputs (0major+65minor)pagefaults 0swaps

但是当我在终端执行 { time ./test.out out 2> exe_time 时，我进入了 exe_time 文件

真正的 0m0.001s
用户 0m0.000s
系统 0m0.000s

如何使用 python 获得第二个版本的输出？

Answer 1

使用bash 调用您的代码，而不是/bin/sh（system() 的默认设置）：

subprocess.Popen(['bash', '-c', '{ time ./test.out < inp;} > out 2> exe_time'])

但是请注意，上面的代码参数化以使用任意文件名是不安全的。一个更好的实践实现可能看起来像：

o, e = subprocess.Popen(['bash', '-c', 'time "$@" 2>&1', '_', './test.out'],
                        stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate()

print("Output from command is:")
sys.stdout.write(o + "\n")
print("Output from time is:")
sys.stdout.write(e + "\n")

注意：

• 我们明确调用 bash，从而确保使用其内置的 time 实现。
• 从 shell 脚本带外传递参数可以安全地将任意参数传递给正在运行的脚本，而不必担心这些参数是否包含尝试的 shell 注入攻击。
• 在 shell 脚本中重定向 2>&1 可确保 test.out 编写的任何 stderr 都与其他输出连接，而不是与 time 命令的输出混合。
• 如果我们确实想要将输出重定向到文件，更好的做法是从 Python 中做到这一点，就像 stdout=open('out', 'w'), stderr=open('exe_time', 'w') 一样。

Answer 2

os.system() 使用/bin/sh。 Bash 有自己的 time 内置函数，而不是 time 二进制文件：

$ /usr/bin/time ls /asd
ls: /asd: No such file or directory
        0.00 real         0.00 user         0.00 sys
$ time ls /asd
ls: /asd: No such file or directory

real    0m0.018s
user    0m0.008s
sys     0m0.013s

如果您想查看执行命令需要多长时间，只需使用subprocess：

import time
import subprocess

with open('inp', 'rb') as input_file:
    with open('out', 'wb') as output_file:
        start = time.time()
        subprocess.call(['./test.out'], stdin=input_file, stdout=output_file)
        runtime = time.time() - start