python 3d列表操作

Question

我有一个3d列表Aa = [[2,3,4,5]，[6、7、8、9]],[[11、12、14、15]]],它由两个二维列表我是怎么得到这个结果的[[2, 6], [11]]每个子列表的第一个元素

b = []
for i, row in enumerate(aa):
for j, rr in enumerate(row):
b.append(rr[0])

给了[2,6,11]

Answer 1

您可以使用列表推导式:

[[i[0] for i in j] for j in aa]

输出:

[[2, 6], [11]]

Answer 2

是存储每个子列表中第一个元素和合并的子列表让你结果

arr这是你的输入数组

ret = []
for twoD in arr:
temp=[]
for oneD in twoD:
temp.append(oneD[0])
ret.append(temp)
print(ret)

输出

[[2, 6], [11]]