لطفاً کد پایتونی که نیاز به کمک دارید را ارائه دهید. مشخص کنید که آیا به راه‌حل، بهینه‌سازی یا توضیح نیاز دارید. من کد را بررسی کرده و بر اساس درخواست شما کمک خواهم کرد.