步函数是AWS提供的一种无服务器计算服务，它可以根据事件触发执行特定的任务。步函数可以定义有序的任务流，每个任务都是一个步骤（Step），并且可以等待一段时间后再继续执行。

下面是一个使用步函数等待功能的示例代码：

import time
import boto3

client = boto3.client('stepfunctions')

def lambda_handler(event, context):
    # 定义步函数的状态机
    state_machine_arn = 'arn:aws:states:us-east-1:123456789012:stateMachine:MyStateMachine'

    # 开始执行步函数
    response = client.start_execution(
        stateMachineArn=state_machine_arn,
        name='MyExecution',
        input='{}'
    )
    
    # 获取步函数执行的ARN
    execution_arn = response['executionArn']
    
    # 等待步函数执行完成
    while True:
        response = client.describe_execution(
            executionArn=execution_arn
        )
        
        status = response['status']
        
        if status == 'SUCCEEDED':
            print('步函数执行成功')
            break
        elif status == 'FAILED':
            print('步函数执行失败')
            break
        elif status == 'RUNNING':
            print('步函数正在执行，等待中...')
        else:
            print('步函数执行状态未知')
        
        time.sleep(5)  # 每5秒检查一次步函数的执行状态
    
    return

上述代码中，首先通过boto3库创建一个步函数客户端对象client。然后，使用start_execution方法来启动一个步函数的执行。获取步函数的ARN后，通过循环调用describe_execution方法来获取步函数的执行状态。如果步函数的执行状态为SUCCEEDED，表示执行成功；如果为FAILED，表示执行失败；如果为RUNNING，表示正在执行，需要继续等待；如果状态未知，则输出未知状态。在每次检查后，通过time.sleep方法等待5秒。

这样，可以在代码中使用步函数等待功能，实现对步函数执行的等待，直到步函数执行完成。