I have a 3d list
aa = [[[2, 3, 4, 5],
[ 6, 7, 8, 9]], [[11, 12, 14, 15]]], which consists of two 2d lists
how do I get this result
[[2, 6], [11]]
the first element of each sub list.
JavaScript
x
5
1
b = []
2
for i, row in enumerate(aa):
3
for j, rr in enumerate(row):
4
b.append(rr[0])
5
gives
[2,6,11]
Advertisement
Answer
You can do it with a list comprehension:
JavaScript
1
2
1
[[i[0] for i in j] for j in aa]
2
Output:
JavaScript
1
2
1
[[2, 6], [11]]
2