创建一个包含解决方案数组的矩阵

Question

我需要在Python中创建一个矩阵，其中包含具有以下形式的未知数组列表：

r_i=[r1,r2,r3,r4,th_12,th_13]

我正在运行带有几个if条件的语句，这些条件会在输出中提供一些我从一开始就不知道的r_i数组。

我正在寻找像 append 这样的函数，我通常用它来创建一个包含我生成的所有解法的向量，但这次每个解决方案都不是单个值而是一个包含6个值的数组我无法生成我想要的东西。

我需要创建一个像这样的矩阵，其中每个r_1都有我上面编写的代码形式。

编辑：我想生成一个numpy数组（R_tot应该是一个numpy数组）。

Answer 1

您可以像我在评论中解释的那样正常生成数组：

r_tot = []

for r_i in however_many_rs_there_are: # each r_i contains an array of 6 values
    r_tot.append(r_i)

然后您可以将r_tot转换为numpy数组，如下所示：

import numpy
np_array = numpy.array(r_tot)

这是一个非常简单的概念证明：

>>> import random, numpy
>>> r_tot = []
>>> for i in range(0,random.randint(1,20)): # append an arbitrary number of arrays
        r_i = [1,2,3,4,5,6]                 # all of size six
        r_tot.append(r_i)                   # to r_tot

>>> np_array = numpy.array(r_tot)           # then convert to numpy array!
>>> np_array                                # did it work?
array([[1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],                  # yeaaaah
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6],
       [1, 2, 3, 4, 5, 6]])