python菱形代码