Optimized Bubble Sort

def swapit(array_number, num):
    count=0
    #print(num)
    #print(array_number)
    for k in range(0,num):
        flag=False
        for j in range(0,num-k-1):
            if(array_number[j]>array_number[j+1]):
                flag=True
                temp=array_number[j]
                array_number[j]=array_number[j+1]
                array_number[j+1]=temp
                count+=1
     
        if(flag==False):
            return count
        print(array_number)
 

print("enter array")
array_num=list(map(int,input().split()))
num=len(array_num)
#print(num)
#print(array_num)

print(swapit(array_num,num))