在python中从df-h输出中选择特定列

Question

我正在尝试创建一个简单的脚本，该脚本将从unixdf - h命令中选择特定的列。我可以使用awk来实现这一点，但是我们如何在python中实现这一点呢？

这里是df -h输出：

Filesystem                    Size  Used  Avail  Use%  Mounted on
/dev/mapper/vg_base-lv_root   28G   4.8G    22G   19%  /
tmpfs                        814M   176K   814M    1%  /dev/shm
/dev/sda1                    485M   120M   340M   27%  /boot

我想要这样的东西：

第1栏：

Filesystem
/dev/mapper/vg_base-lv_root           
tmpfs                 
/dev/sda1

第2栏：

Size
28G
814M 
485M   

Answer 1

下面是完整的示例：

import subprocess
import re

p = subprocess.Popen("df -h", stdout=subprocess.PIPE, shell=True)
dfdata, _ = p.communicate()

dfdata = dfdata.replace("Mounted on", "Mounted_on")

columns = [list() for i in range(10)]
for line in dfdata.split("\n"):
    line = re.sub(" +", " ", line)
    for i,l in enumerate(line.split(" ")):
        columns[i].append(l)

print columns[0]

它假定装入点不包含空格。

下面是一个更完整（更复杂）的解决方案，它不需要硬核数列：

import subprocess
import re

def yield_lines(data):
    for line in data.split("\n"):
        yield line

def line_to_list(line):
    return re.sub(" +", " ", line).split()

p = subprocess.Popen("df -h", stdout=subprocess.PIPE, shell=True)
dfdata, _ = p.communicate()

dfdata = dfdata.replace("Mounted on", "Mounted_on")

lines = yield_lines(dfdata)

headers = line_to_list(lines.next())

columns = [list() for i in range(len(headers))]
for i,h in enumerate(headers):
    columns[i].append(h)

for line in lines:
    for i,l in enumerate(line_to_list(line)):
        columns[i].append(l)

print columns[0]