Объединение двух числовых списков python по определенному принципу

Нужно объединить два числовых списка. Нужно их объединить по принципу а[1,1] b[2,2], а и b [1,2,1,2]. Помогите оптимизировать код в функции.

a = [int(s) for s in input().split()]
b = [int(s) for s in input().split()]
def combine_lists(a, b):
​ ​ ​ ​ len_a = len(a)
​ ​ ​ ​ len_b = len(b)
​ ​ ​ ​ if len_a < len_b:
​ ​ ​ ​ ​ ​ ​ ​ limit = len_a
​ ​ ​ ​ else:
​ ​ ​ ​ ​ ​ ​ ​ limit = len_b
​ ​ ​ ​ i = 0
​ ​ ​ ​ r = []
​ ​ ​ ​ while i < limit:
​ ​ ​ ​ ​ ​ ​ ​ r.append(a[i])
​ ​ ​ ​ ​ ​ ​ ​ r.append(b[i])
​ ​ ​ ​ ​ ​ ​ ​ i += 1
​ ​ ​ ​ return r

if __name__ == '__main__':
​ ​ ​ ​
​ ​ ​ ​ print(repr(combine_lists(a, b)))