python - 过滤一个 numpy meshgrid

Question

我想过滤一个 numpy meshgrid 的值:

X,Y = np.mgrid[-10:10,-10:10]

在这种情况下，我想删除 x**2 + y**2 <= 2 的所有坐标.但是，当我尝试直接过滤数组时，例如

filter(lambda x,y: x**2 + y**2 >= 2, np.meshgrid[-10:10,-10:10])

我收到错误是因为我没有正确处理数组的结构。

任何正确执行此操作的提示都将不胜感激!

Answer 1

我能够使用 numpy.where 实现我需要的结果，方法是分别过滤每个数组，但在 where 条件中引用它们:

X,Y = np.mgrid[-10:10,-10:10]
X,Y = np.where(X**2 + Y**2 > 2, X, 0), np.where(X**2 + Y**2 > 2, Y, 0)

这会产生新的二维数组，这是我对 matplotlib 所需要的。感谢所有花时间看这个问题的人!