python - 以矩阵形式将列表排列到numpy数组

Question

我有一个列表:

rs = list([[(0, 167, 159, 0), (1, 168, 160, 1)], [(0, 167, 159, 1), (1, 168, 160, 0)]])

我想把它排列成矩阵形式的numpy数组，所以我写了下面的代码:

import numpy

rs = list([[(0, 167, 159, 0), (1, 168, 160, 1)], [(0, 167, 159, 1), (1, 168, 160, 0)]])
k = 2
j = 0
h = 0
ars = []
element = list(rs)
ck_dist_m = numpy.array([[]], ndmin = 2)

while j <= k - 1:
    while h <= k - 1:
        rp = numpy.array(element[j][h][3], ndmin = 2)
        ars.append(rp)
        h = h + 1
    else:
        h = 0
        ck_dist_m = numpy.vstack(ars)
    j = j + 1
print ck_dist_m

我想要的结果是:

[[0, 1],
 [1, 0]]

但是上面的脚本给出了这个:

[[0]
 [1]
 [1]
 [0]]

如何编辑脚本得到矩阵形式的结果？

Answer 1

In the while loop I extract the third element in the sublist, and try to store them in matrix form.

这里有一个更直接的方法来实现这一点:

>>> numpy.array(rs)[:,:,3]
array([[0, 1],
       [1, 0]])