b=[b for b in input().split()] ну вот вырвиглаз с именами
вот это
for i in range(len(b)):
a.append(b[i])
надо записать в предыдущем варианте как максимум, а как минимум просто for i in b
ну и в целом конструкция простая и понятная, но понятная только вам )))