从列表列表数据结构中检索元素

如何解决从列表列表数据结构中检索元素

我正在运行以下代码段

import numpy as np
np.random.seed(123)
from skopt.space import Space
from skopt.sampler import Lhs
space = Space([(0.,1.),(0.,1.)])
lhs = Lhs(lhs_type="centered",criterion=None)
theta_design = lhs.generate(space.dimensions,200)

得到的 theta_design 是一个长度为 200 的列表。每个列表元素也是一个长度为 3 的列表。部分结果如下所示。所以 theta_design 是一种大小为 200*3 的矩阵。我的问题是如何沿此列表列表结构的第一维检索元素。换句话说，我想从这个矩阵中检索前 100 个元素。

解决方法

first_col,second_col,third_col = zip(*theta_design[:100])

,

我认为您想要获取前 100 个列表中每个列表的第一个元素，并将它们保存到列表中。 这可以通过 list comprehension 完成。

first_element_of_each_list = [row[0] for row in theta_design[:100]]