该代码报错，希望得到指正。

该代码显示 TypeError: only size-1 arrays can be converted to Python scalars 希望的到指正。