如何为 Python 代码片段添加标题

Question

我正在使用自定义命令为我想在文档中显示的 Python 代码创建环境。此命令取自 redmode 对以下问题的回答：这里。

所以，我想知道：有没有办法将标题或子标题包含到此代码片段中？ lstlisting 环境允许此选项，我将不胜感激。