2013-07-24 64 views
1

任何人都可以帮助我 - 是否有任何方式来创建像这样的numpy数组坐标网格?创建和使用坐标网格

(0,0) (0,1) (0,2) ... (0,n) 
(1,0) (1,1) (1,2) ... (1,n) 
........................... 
(m,0) (m,1) (m,2) ... (m,n) 

如果是的话,我怎样才能找到从每个点到中心在(m/2,n/2)和半径R的圆的距离?

(x - m/2)^2 + (y - n/2)^2 - R^2 = ? 
+0

你可以查看http://stackoverflow.com/questions/17663393了解相似的例子。 – Daniel

+0

好点,这两个问题非常相似。 – IanH

回答

4

这样做的标准方法是使用meshgrid函数。它使两个数组与您想要的点的坐标为xy。要在你的问题得到显示的坐标,你可以做

import numpy as np 
x = np.arange(m+1) 
y = np.arange(n+1) 
X, Y = np.meshgrid(x, y) 

然后计算出你愿意,你可以做

np.sqrt((X - m/2.)**2 + (Y - n/2.)**2) - R 

欲了解更多信息的距离上meshgrid看到的文档 http://docs.scipy.org/doc/numpy/reference/generated/numpy.meshgrid.html

而且如果您希望在两个端点之间使用均匀间隔的值而不是0m0n,考虑使用linspace函数。

+1

由于它看起来像OP是新的numpy,我想添加如何实际获得坐标。如果让'idx = np.sqrt((X - m/2。)** 2 +(Y - n/2。)** 2 - R ** 2)','idx'现在是一个True/False的值与'X'的大小相同。因此'X [idx]'和'Y [idx]'只返回圆圈中的坐标。 – Hooked

+0

谢谢,这帮助我注意到了解决方案中的一个错字,我只是修复了它。我想你的意思是设置'idx =((X - m/2。)** 2 +(Y - n/2。)** 2) IanH