在linux中使用python创建一个像pstree命令的进程树

发布时间：2020-12-13 22:54:51 所属栏目：Linux 来源：网络整理

导读：我是 python的新手.我想编写一个在stdout上输出树状图形的程序. 我理想的输出是： 0|__0|__4| |__360| |__1000272|__3460 我收集的数据如下： 0 : [0,4]4 : [360]272 : [3460]368 : [4184]472 : [504,576,7016]568 : [584,640]576 : [664,672]640 : [1048]66

我是 python的新手.我想编写一个在stdout上输出树状图形的程序.
我理想的输出是：

0
|__0
|__4
|  |__360
|      |__1000
272
|__3460

我收集的数据如下：

0       : [0,4]
4       : [360]
272     : [3460]
368     : [4184]
472     : [504,576,7016]
568     : [584,640]
576     : [664,672]
640     : [1048]
664     : [368,372,512,788]
788     : [2120,2720,2976,2996,3956,3980]

左列是父进程id,右列是子进程id.
我把数据放在一个名为dic的字典中.因此,字典键是父进程id,字典值是由子进程ID组成的列表.

我的代码是这样的：

for key in dic.keys():
    print key,'n|'
    for v in dic[key]:
        print '__',v,'n|'

问题是我只能输出两层树.以数据为例,576作为父id也是472的子id.因此576,664,672应该放在472的子树中.
我的代码对此不起作用.看来我们需要使用递归函数.
但我不知道如何处理它.

你们能给我一些提示吗？

编辑：
根据我收集的数据,有一些父ID没有祖父母.
所以最终的产量应该是森林.不是单根的树.

解决方法

这个怎么样：

def printTree(parent,tree,indent=''):
  print parent
  if parent not in tree:
    return
  for child in tree[parent][:-1]:
    sys.stdout.write(indent + '|-')
    printTree(child,indent + '| ')
  child = tree[parent][-1]
  sys.stdout.write(indent + '`-')
  printTree(child,indent + '  ')

tree = {
  0       : [0,4],4       : [360],272     : [3460],368     : [4184],472     : [504,7016],568     : [584,640],576     : [664,672],640     : [1048],664     : [368,788],788     : [2120,3980]
}

printTree(472,tree)

printTree(472,tree)
472
|-504
|-576
| |-664
| | |-368
| | | `-4184
| | |-372
| | |-512
| | `-788
| |   |-2120
| |   |-2720
| |   |-2976
| |   |-2996
| |   |-3956
| |   `-3980
| `-672
`-7016

也许这就是你喜欢它的方式,我不知道.

它没有为递归内置任何检查,所以如果你在0上尝试它,它将进入无限递归(并最终由于堆栈溢出而中止).您可以通过传递已处理节点的跟踪来自行检查递归.

这也没有在您的林中找到树根列表,因此您也必须这样做. (但这听起来像另一个问题.)

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!