我正在尝试使用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
位置元素的最佳方法是什么?
答案 0 :(得分:3)
你非常接近,只需要从xaxis
和yaxis
创建一个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)