写一篇基于python的俄罗斯方块代码

好的，以下是一个基于Python的俄罗斯方块代码示例：

import pygame
import random

# 初始化pygame
pygame.init()

# 设置屏幕大小和标题
screen = pygame.display.set_mode((500, 600))
pygame.display.set_caption("俄罗斯方块")

# 定义方块大小和颜色
block_size = 30
colors = [(255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0), (255, 0, 255), (0, 255, 255), (128, 128, 128)]

# 定义方块形状
shapes = [
    [[1, 1, 1],
     [0, 1, 0]],

    [[0, 2, 2],
     [2, 2, 0]],

    [[3, 3, 0],
     [0, 3, 3]],

    [[4, 0, 0],
     [4, 4, 4]],

    [[0, 0, 5],
     [5, 5, 5]],

    [[6, 6, 6, 6]],

    [[7, 7],
     [7, 7]]
]

# 定义方块初始位置和速度
block_x, block_y = 4, 0
block_speed = 1
block_shape = random.choice(shapes)
block_color = random.choice(colors)

# 定义游戏循环
running = True
while running:

    # 画出游戏界面
    screen.fill((0, 0, 0))
    pygame.draw.rect(screen, (255, 255, 255), (50, 50, 300, 500), 5)
    pygame.draw.rect(screen, (255, 255, 255), (360, 50, 100, 100), 5)
    pygame.draw.rect(screen, (255, 255, 255), (360, 200, 100, 100), 5)
    pygame.draw.rect(screen, (255, 255, 255), (360, 350, 100, 100), 5)

    # 画出当前方块
    for i in range(len(block_shape)):
        for j in range(len(block_shape[i])):
            if block_shape[i][j] != 0:
                pygame.draw.rect(screen, block_color, (50 + (block_x + j) * block_size, 50 + (block_y + i) * block_size, block_size, block_size))

    # 控制方块下落速度
    if pygame.time.get_ticks() % 500 == 0:
        block_y += block_speed

    # 检测方块是否到达底部或碰到其他方块
    for i in range(len(block_shape)):
        for j in range(len(block_shape[i])):
            if block_shape[i][j] != 0 and (block_y + i + 1 == 20 or grid[block_y + i + 1][block_x + j] != 0):
                grid[block_y + i][block_x + j] = block_color
                block_x, block_y = 4, 0
                block_speed = 1
                block_shape = random.choice(shapes)
                block_color = random.choice(colors)
                break

    # 更新方块所在的行和列
    rows, columns = len(grid), len(grid[0])

    # 画出已经填充的方块
    for i in range(rows):
        for j in range(columns):
            if grid[i][j] != 0:
                pygame.draw.rect(screen, grid[i][j], (50 + j * block_size, 50 + i * block_size, block_size, block_size))

    # 检测是否有行被填满
    for i in range(rows):
        if all(grid[i]):
            grid.pop(i)
            grid.insert(0, [0] * columns)

    # 检测游戏是否结束
    if any(grid[0]):
        running = False

    # 处理事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT and block_x > 0:
                if all(grid[block_y + i][block_x + j - 1] == 0 for i in range(len(block_shape)) for j in range(len(block_shape[i])) if block_shape[i][j] != 0):
                    block_x -= 1
            elif event.key == pygame.K_RIGHT and block_x < 10 - len(block_shape[0]):
                if all(grid[block_y + i][block_x + j + 1] == 0 for i in range(len(block_shape)) for j in range(len(block_shape[i])) if block_shape[i][j] != 0):
                    block_x += 1
            elif event.key == pygame.K_DOWN and block_speed < 5:
                block_speed += 1

    # 刷新屏幕
    pygame.display.flip()

# 退出pygame
pygame.quit()

这个代码实现了一个基本的俄罗斯方块游戏，其中包括方块的移动和旋转、方块的下落和堆积、已经填充的方块的消除等功能。你可以根据自己的需求对代码进行修改和完善。