import
math
def
rNN(points, p, r
=
2.2
):
freq1
=
0
freq2
=
0
for
group
in
points:
for
feature
in
points[group]:
if
math.sqrt((feature[
0
]
-
p[
0
])
*
*
2
+
(feature[
1
]
-
p[
1
])
*
*
2
) <
=
r:
if
group
=
=
0
:
freq1
+
=
1
elif
group
=
=
1
:
freq2
+
=
1
return
0
if
freq1>freq2
else
1
def
main():
points
=
{
0
:[(
1.5
,
4
), (
1.8
,
3.8
), (
1.65
,
5
), (
2.5
,
3.8
), (
3.8
,
3.8
)],
1
:[(
5.5
,
3.5
), (
5.6
,
4.5
), (
6
,
5.4
), (
6.2
,
4.8
), (
6.4
,
4.4
)]}
p
=
(
4.5
,
4
)
r
=
2.2
print
(
"The value classified to query point is: {}"
.
format
(
rNN(points, p, r)))
if
__name__
=
=
'__main__'
:
main()