python—引用类型的巧用

• 先有一个列表，装了很多字典，我们希望所有parent_id不为None的都将其放到对应id的键children下，并递归执行，实现一个多级字典，但递归是比较耗内存的，我们将使用引用类型的特殊方法来实现。

# 原始列表
comment_list = [
    {'id': 1, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': None},
    {'id': 2, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': None},
    {'id': 3, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': None},
    {'id': 4, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': 2},
    {'id': 5, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': 4},
    {'id': 6, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': 3},
    {'id': 7, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': None},
    {'id': 8, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': 5},
    {'id': 9, 'user': '杀杀杀', 'comment': '我丢', 'parent_id': 8},
]
# 最后存放结果的列表
comments = []	
# 将列表转化为字典，并用id作为comment_dict的键和添加一个children键值对
comment_dict = {}	
for _ in comment_list:
    _.update({'children': []})
    comment_dict[_['id']] = _

• 转化为字典之后的结果：

print(comment_dict)
"""
{
	1: {
		'id': 1,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': None,
		'children': []
	},
	2: {
		'id': 2,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': None,
		'children': []
	},
	3: {
		'id': 3,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': None,
		'children': []
	},
	4: {
		'id': 4,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': 2,
		'children': []
	},
	5: {
		'id': 5,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': 4,
		'children': []
	},
	6: {
		'id': 6,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': 3,
		'children': []
	},
	7: {
		'id': 7,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': None,
		'children': []
	},
	8: {
		'id': 8,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': 5,
		'children': []
	},
	9: {
		'id': 9,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': 8,
		'children': []
	}
}
"""

• 接下来修改这个字典中有children值的项，将对应的children加入到键值对中，

for v in comment_dict.values():
    if not v['parent_id']
    	# 将parent_id为None，即这里只放入所有第一级元素
        comments.append(v)
    else:
    	"""注意在这里只修改了原来字典中的children值"""
        comment_dict[v['parent_id']]['children'].append(v)
        

print(comments)

当我们打印comments时会发现已经实现了多级字典

[{
	'id': 1,
	'user': '杀杀杀',
	'comment': '我丢',
	'parent_id': None,
	'children': []
}, {
	'id': 2,
	'user': '杀杀杀',
	'comment': '我丢',
	'parent_id': None,
	'children': [{
		'id': 4,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': 2,
		'children': [{
			'id': 5,
			'user': '杀杀杀',
			'comment': '我丢',
			'parent_id': 4,
			'children': [{
				'id': 8,
				'user': '杀杀杀',
				'comment': '我丢',
				'parent_id': 5,
				'children': [{
					'id': 9,
					'user': '杀杀杀',
					'comment': '我丢',
					'parent_id': 8,
					'children': []
				}]
			}]
		}]
	}]
}, {
	'id': 3,
	'user': '杀杀杀',
	'comment': '我丢',
	'parent_id': None,
	'children': [{
		'id': 6,
		'user': '杀杀杀',
		'comment': '我丢',
		'parent_id': 3,
		'children': []
	}]
}, {
	'id': 7,
	'user': '杀杀杀',
	'comment': '我丢',
	'parent_id': None,
	'children': []
}]