如何获得均匀间隔阵列的位置坐标(n×2)?

时间:2016-07-18 11:14:16

标签: python arrays numpy psychopy

我正在尝试使用PsychoPy的ElementArrayStim制作一个线元素数组(23×23网格)。

对于线元素位置的xys参数,我试图以统一的方式(23×23网格)定位线元素。

我试图通过执行以下操作来获取元素的位置:

nx, ny = (23, 23)
xaxis = np.linspace(-220, 220, nx)
yaxis = np.linspace(-220, 220, ny)

yx = np.meshgrid(xaxis, yaxis)

我收到的输出似乎是2个独立的数组(我假设x轴坐标和y轴坐标),但它们似乎按每行列出。

但是,PsychoPy只接受'xys参数的n×2输入 - 我不知道如何改变输出的形状,因此它的形式为n×2。

另外,如果我使用的方法不正确/效率低,那么在n×2形状中实现xys位置元素的最佳方法是什么?

  • 乘以n×2,我的意思是两列,23×23 = 529行。列将分别用于x和y坐标,并且每个元素将包含529行。

2 个答案:

答案 0 :(得分:3)

你非常接近,只需要从xaxisyaxis创建一个3D坐标数组,然后根据需要重塑该3D数组以获得529行×2列2D数组:

In [21]: xy = np.dstack(np.meshgrid(xaxis, yaxis)).reshape(-1, 2)

In [22]: xy
Out[22]: 
array([[-220, -220],
       [-200, -220],
       [-180, -220],
       ..., 
       [ 180,  220],
       [ 200,  220],
       [ 220,  220]])

In [23]: xy.shape
Out[23]: (529L, 2L)

或者,您可以通过以下方法获得相同的结果:

xy = np.mgrid[-220:240:20, -220:240:20].T.reshape(-1, 2)

答案 1 :(得分:0)

你是否需要遍历两个轴,在你去的时候将每个元素附加到数组?

fred=np.array([[],[]])
for column in xaxis:
    for row in yaxis:
        fred = np.append(fred, [[column], [row]], 1)

fred.shape
(2, 529)