Python 要怎麼利用Function函式寫出數字三角形

個人目前是有寫出一個輸出符合他要求的程式碼
但是後來才發現他需要使用函式來寫
但是我函式實在不太了解
實在沒辦法
求解各位前輩幫忙小弟

題目要求

輸出結果

小弟自己寫的程式碼

 
q =int(input())
if q==1:
    for i in range(1,int(input())+1): 
        print(pow(int(pow(10,i)//9),2))
if q==2:
    w =int(input())
    for x in range(1,w+1):
        print('_'*(w-x),end='')
        n=1
        while n<=x:
            print(n,sep='',end='')
            n+=1
        n=x-1
        while n>=1:
            print(n,sep='',end='')
            n-=1
        print()
if q==3:
    w= int(input())
    for i in range(w,0,-1):
        print('_'*(w-i),end='')
        for j in range(1,i+1):
            print(j,end='')
        for j in range(i-1,0,-1):
            print(j,end='')
        print()