如何在存储在DataFrame中的树中查找叶子

import pandas as pd

# 创建一个示例DataFrame，表示树结构
data = {'节点ID': [1, 2, 3, 4, 5, 6, 7],
        '父节点ID': [None, 1, 1, 2, 2, 3, 3]}
df = pd.DataFrame(data)

# 存储叶子节点的列表
leaves = []

# 递归函数，查找叶子节点
def find_leaves(node_id):
    # 查询当前节点的子节点
    children = df[df['父节点ID'] == node_id]
    
    # 如果当前节点没有子节点，将其添加到叶子节点列表
    if len(children) == 0:
        leaves.append(node_id)
    else:
        # 递归遍历子节点
        for child_id in children['节点ID']:
            find_leaves(child_id)

# 从根节点开始查找叶子节点
root_id = df[df['父节点ID'].isnull()]['节点ID'].values[0]
find_leaves(root_id)

# 打印叶子节点
print("叶子节点：", leaves)