Python - Display SSID & choose option with number input

Question

I'm using Python to display SSID networks within range and then able to choose which one of these I'd like to connect to. The code asks for input to which SSID it should connect but it needs the full name of the desired network. Is there anyway i could change this so that i would only need to input a number for a desired network? Example: Input '2' would connect to the second listed network in the terminal.

I will insert my code below to show what I'm working with so far. I'm aware of other flaws in the code that I will come to fix later on, as of now im focused on the aspect of changing the way i input the desired network.

Cheers!

# function to establish a new connection
def createNewConnection(name, SSID, password):
    config = """<?xml version=\"1.0\"?>
<WLANProfile xmlns="http://www.microsoft.com/networking/WLAN/profile/v1">
    <name>"""+name+"""</name>
    <SSIDConfig>
        <SSID>
            <name>"""+SSID+"""</name>
        </SSID>
    </SSIDConfig>
    <connectionType>ESS</connectionType>
    <connectionMode>auto</connectionMode>
    <MSM>
        <security>
            <authEncryption>
                <authentication>WPA2PSK</authentication>
                <encryption>AES</encryption>
                <useOneX>false</useOneX>
            </authEncryption>
            <sharedKey>
                <keyType>passPhrase</keyType>
                <protected>false</protected>
                <keyMaterial>"""+password+"""</keyMaterial>
            </sharedKey>
        </security>
    </MSM>
</WLANProfile>"""
    command = "netsh wlan add profile filename=\""+name+".xml\""+" interface=Wi-Fi"
    with open("testnamn.xml", 'a') as file: #open for append insread
        file.write(config)
    os.system(command) #save to another folder

# function to connect to a network
def connect(name, SSID):
    command = "netsh wlan connect name=\""+name+"\" ssid=\""+SSID+"\" interface=Wi-Fi"
    os.system(command)

# function to display avavilabe Wifi networks
def displayAvailableNetworks():
    command = "netsh wlan show networks interface=Wi-Fi"
    os.system(command)

# display available networks
displayAvailableNetworks()

# input wifi name and password
name = input("Name of Wi-Fi: ")
password = input("Password: ")