I have the following two lists:
lista = ['a', 'b']
listb = ['c', 'd']
And I want the output list like this:
res = [['a', 'c'], ['b', 'd']]
My solution is
arr = np.array([lista, listb]).T
res = arr.tolist()
Is there a simpler way to do it?
2 Answers
[list(a) for a in zip(lista, listb)]
Whether this is simplier might be subjective, but it is one of the options for sure.
3 Comments
accdias
Isn't easier to just do
list(zip(lista, listb))?
matszwecja
@accdias No, because then inner objects will be tuples and not lists.
accdias
Oh! I see your point. :-D
You can declare res variable and pass both lists as arguments. It is a lot simpler. Also, don't use list as a variable name as it is a built-it function name
res = []
for i in range(len(lista)):
temp = [lista[i], listb[i]]
res.append(temp)
You can also use list comprehension which is faster and more transparent
res = [[lista[x], listb[x]] for x in range(len(lista))]
3 Comments
matszwecja
This is not what OP wants.
matszwecja
"And I want the output list like this: res = [['a', 'c'], ['b', 'd']]"
Aleksander Ikleiw
@matszwecja my bad, edited and posted correct solution
listb=['b', 'd', 'f']should the result be[['a', 'c'], ['b', 'd'], ['f']]or[['a', 'c'], ['b', 'd'], ['', 'f']]?['a', 'c', 'b', 'd'], you just need to use the+operator:res = lista + listb.