get item from index_in sublist

a=[[1,2,3,4],[5,6,7]]
b_index=[1,0]

lst=[]

for ls,indx in zip(a,b_index):
    sub=[ls[indx]]
    lst.append(sub)

OUT=lst