从嵌套列表中生成列表的更简单方法？

Question

使用一些数值算法，我的代码生成一个矩阵列表，它存储在一个嵌套列表中，如下所示

A = [matrix([[1,2],[3,4]]), matrix([[5,6],[7,8]]), ...)

随后，我想将值1,5,9，......与其他列表相对应，比如说＆＃39; x＆＃39;，长度相同。目前我循环遍及我想要的值

wanted_sol
for i in range(0,len(A))
   wanted_sol.append(A[i][0,0])

然后我绘制了＆＃39; wanted_sol＆＃39;。我想知道是否有更短的方法来做到这一点？我尝试了几件事，比如

plot(x, A[:][0,0])
plot(x, A[0:len(A)][0,0]),

但我无法让它发挥作用。

Answer 1

您可以将A转换为Baseclass obj = new Derivedclass(). 并使用numpy切片表示法：

numpy.ndarray