Pygame already installed; however, python terminal says "No module named 'pygame' " (Ubuntu 20.04.1)

I'm using Ubuntu 20.04.1 LTS. I have installed pygame with sudo apt-get install python-pygame. However, when I run my code using VS code, the python terminal gives the output below:

Traceback (most recent call last):
  File "path", line 1, in <module>
    import pygame
ModuleNotFoundError: No module named 'pygame'

I tried to reinstall Pygame using the same command. Here is the output:

Reading package lists... Done
Building dependency tree       
Reading state information... Done
python-pygame is already the newest version (1.9.6+dfsg-2build1).
0 upgraded, 0 newly installed, 0 to remove and 5 not upgraded.

Basically, I already have pygame installed. So what do you think might be the reason? My Python version is 3.8.2.

Thanks in advance!