python - 按数组名称追加数组

Question

我正在使用一个函数，它输出一个大约 600 个元素长的 numpy 数组。

array = function_output()
array.shape # this outputs (600,)

我必须处理这个函数的大约 50 个输出。每个输出都是不同的。目标是将这些数组中的每一个连接在一起形成一个列表(或 numpy 数组)，然后保存结果。

我愚蠢地开始单独标记每个数组，即

array1 = function_output()
array2 = function_output() 
...

然后我想我可以将其连接到一个列表中，即

list = [array1, array2, array3, ..., array50]

(1) 我认为一开始没有任何办法可以通过命名方案来解决。函数的每个输出都是唯一的，应适当标记

(2) 然而，通过复制和粘贴的方式定义列表感觉很愚蠢。我可以以某种方式使用“for”语句来迭代变量名称吗？

Answer 1

每当您有编号变量名称时，请考虑使用列表:

output = [function_output() for i in range(50)]

您可以使用 output[0] 而不是使用 array1 访问第一个数组(因为 Python 使用基于 0 的索引。)

要将数组列表合并到一个 NumPy 数组中，您可以使用

array = np.column_stack(output)

array.shape 将是 (600, 50)。