用python写俄罗斯方块代码

好的，下面是一个简单的俄罗斯方块游戏代码示例，你可以参考一下：

import pygame
import random

# 初始化 pygame
pygame.init()

# 游戏区域的大小
screen_width = 300
screen_height = 600

# 方块的大小
block_size = 30

# 方块的颜色
colors = [(0, 0, 0), (255, 0, 0), (0, 255, 0), (0, 0, 255), (255, 255, 0)]

# 定义方块类
class Block:
    def __init__(self, x, y, color):
        self.x = x
        self.y = y
        self.color = color

    def draw(self):
        pygame.draw.rect(screen, self.color, (self.x * block_size, self.y * block_size, block_size, block_size))

# 定义方块组类
class BlockGroup:
    def __init__(self, x, y, shape):
        self.x = x
        self.y = y
        self.shape = shape
        self.color = random.randint(1, len(colors)-1)

    def move_down(self):
        self.y += 1

    def move_left(self):
        self.x -= 1

    def move_right(self):
        self.x += 1

    def rotate(self):
        self.shape = list(zip(*self.shape[::-1]))

    def draw(self):
        for i, j in self.shape:
            Block(self.x + i, self.y + j, colors[self.color]).draw()

# 创建一个窗口
screen = pygame.display.set_mode((screen_width, screen_height))

# 游戏主循环
while True:
    # 处理事件
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            exit()

    # 绘制背景
    screen.fill((255, 255, 255))

    # 绘制方块
    BlockGroup(5, 0, [(0, 0), (0, 1), (1, 0), (1, 1)]).draw()

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

这是一个非常简单的示例代码，只实现了一个方块组的绘制。你可以根据自己的需求，添加更多功能，比如移动、旋转、碰撞检测等。