将嵌套数组列表转为numpy数组后，如何提取第一列元素用于绘图？

你的原始数据是三层嵌套结构：外层是列表，每个子元素是包含一个二维numpy数组的单元素列表。转成numpy数组PN后，它的维度是(n, 1, 2)（n是元素个数，这里为9），所以PN[:,0]只能取出每个子列表里的二维数组，得到形状为(n,2)的数组，还需要进一步提取第一个元素。

以下是几种可行的解决方法：

方法一：直接多维索引 #

利用numpy的多维索引直接定位到目标元素：

first_col = PN[:, 0, 0]

这个操作会直接取出每个嵌套结构里的第一个数值，得到形状为(n,)的一维数组，正好满足绘图需求。

方法二：先压缩冗余维度再提取 #

先用squeeze()去掉数组中长度为1的冗余维度，再提取第一列：

# 压缩后数组形状变为(n, 2)
PN_squeezed = PN.squeeze()
# 提取第一列
first_col = PN_squeezed[:, 0]

方法三：在列表阶段直接处理（更高效） #

如果还没将原始列表转成numpy数组，也可以直接通过列表推导式提取后再转成数组：

first_col = np.array([item[0][0] for item in params])

你可以根据自己的代码场景选择任意一种方法，最终得到的first_col就是所有子数组第一个元素组成的数组，可直接用于绘图。

内容的提问来源于stack exchange，提问作者Khushal