how to set up pygame

python3 -m pip install -U pygame --user
Wha?

pip install pygame

import pygame
import os
import sys
import random
pygame.init()
screen = pygame.display.set_mode((800, 500))