下一步，我继续在注释中写明我需要做到哪些事情。比方说我想继续写一个状态转移方程，用来求马尔科夫链模型的概率矩阵。

可以看到，Amazon CodeWhisperer 直接提供好了一个初始矩阵给我，非常的方便。

我们再进一步提出要求，注释中写清楚即可。

补全之后下一步 Amazon CodeWhisperer 又根据新的注释帮我进行了代码建议，是不是看起来很专业，而且代码风格看起来赏心悦目，非常整齐，看起来清爽。

然后我们简单对状态矩阵进行判断返回即可。根据 Amazon CodeWhisperer 的代码建议补全如下。