写一段python代码