python - 使用 For 循环结果在 Python 中创建二维数组

Question

我在 for 循环中运行一些代码。我想从我的循环中获取结果并将它们放在一个二维数组中，该数组具有 2 列和与迭代循环次数一样多的行。这是我所拥有的简化版本:

 for i in range(10):

      'bunch of stuff'

      centerx = xc
      centery = yc

如何将 centerx 和 centery 的值保存在 2 列 10 行的二维数组中？感谢任何帮助，谢谢!

Answer 1

你可以试试这个:

import numpy as np
listvals = []
for i in range(10):
    listvals.append((xc, yc))
mat_vals = np.vstack(listvals)

这将输出一个像这样的 ndarray:

[[ 2  0]
 [ 3  1]
 [ 4  2]
 [ 5  3]
 [ 6  4]
 [ 7  5]
 [ 8  6]
 [ 9  7]
 [10  8]
 [11  9]]

或者这可能更好:

import numpy as np
list_xc = []
list_yc = []
for i in np.arange(10): 
    list_xc.append(xc)
    list_yc.append(yc)
mat_xc = np.asarray(list_xc)
mat_yc = np.asarray(list_yc)
mat_f = np.column_stack((mat_xc, mat_yc))