Python两个数组，获取半径内的所有点

我有两个数组，假设 x 和 y 包含几千个数据点。
绘制散点图可以很好地表示它们。现在我想选择一定半径内的所有点。例如 r=10

我试过这个，但它不起作用，因为它不是网格。

x = [1,2,4,5,7,8,....]
y = [-1,4,8,-1,11,17,....]
RAdeccircle = x**2+y**2
r = 10

regstars = np.where(RAdeccircle < r**2)

这与 nxn 数组不同，RAdeccircle = x**2+y**2 似乎不起作用，因为它不会尝试所有排列。

最佳答案

您只能对 numpy 数组执行 **，但在您的情况下，您使用的是列表，并且在列表上使用 ** 会返回错误，因此您首先需要使用 np.array() 将列表转换为 numpy 数组

import numpy as np


x = np.array([1,2,4,5,7,8])
y = np.array([-1,4,8,-1,11,17])
RAdeccircle = x**2+y**2

print RAdeccircle

r = 10

regstars = np.where(RAdeccircle < r**2)
print regstars

>>> [  2  20  80  26 170 353]
>>> (array([0, 1, 2, 3], dtype=int64),)

关于Python两个数组，获取半径内的所有点，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/30714088/