python 逐行循环 subprocess.check_output

Question

我需要遍历命令的输出。我以为我会使用subprocess.check_output，现在我有两个问题。

这是一个文件：

foo
bar

这是我的 python 脚本：

import subprocess

for line in subprocess.check_output(['cat', 'foo']):
  print "%r" % line

这就是我得到的：

$ python subp.py 
'f'
'o'
'o'
'\n'
'b'
'a'
'r'
'\n'

我希望：

$ python subp.py
'foo\n'
'bar\n'

Answer 1

subprocess.check_output(['cat', 'foo']) 返回一个字符串："foo\nbar"

因此，您的 for 循环遍历字符串，逐个打印每个字符。

以下应该可以解决您的问题：

import subprocess

print subprocess.check_output(['cat', 'foo'])

你也可以这样做：

import subprocess

for line in subprocess.check_output(['cat', 'foo']).split('\n'):
    print "%r" % line

Answer 2

使用 Python3 之前的答案不会立即起作用，因为 bytes 由 check_output 返回

然后您可以将字节解码为字符串或立即将它们拆分：

output = subprocess.check_output(['cat', 'foo'])

# splitting with byte-string
for line in output.split(b'\n'): 
    print(line)

# or decoding output to usual string
output_str = output.decode()
for line in output_str.split('\n'):
    print(line)

Answer 3

Python3：

for line in subprocess.check_output(['cat', 'foo'], universal_newlines=True).split('\n'):
    print(line)