问题描述
我的任务:绘制一个初始点并计算新的点位置 (x,y),而点在一个圆圈中。
我的问题: 使用 while
循环执行此操作无法运行代码块,因为它错误地确定圆不包含该点(图 1)。然而,使用 for
循环可以看到多个点驻留在圆中,但仍然错误地识别出圆中不存在点(图 2)。
附加信息:我似乎也无法获得要绘制的线而不是点(例如使用 pyplot.plot(x,y,'-')
)。
如果有更简化的方法,例如绘制圆/点/线,或整个方法,我会很想看到它们。感谢您提供的任何帮助。
# IMPORT LIBRARIES ####
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.path as mPath
import matplotlib.patches as mPatches
# CREATE CIRCLE ####
# Input circle radius
circleRadius = 2
# Create circle (arguments: centre,radius,face colour (fc),edge colour (ec),alpha)
circle = mPatches.Circle((0,0),radius = circleRadius,fc = 'm',alpha = 0.3)
# Plot circle
fig1,ax = plt.subplots()
ax.add_patch(circle)
ax.set_xlim(-5,5)
ax.set_ylim(-5,5)
ax.set_aspect('equal')
# CREATE POINTS ####
# Input coordinates
x0 = -1
y0 = 0.5
# Assignment for equation
x = x0
y = y0
# Store
curr_pt = [x,y]
counter = 0
# Plot 1st point (x,y)
plt.plot(x,'.c')
# Pre-checks
print(circle.contains_point([x,y]))
print(counter)
print(x0,y0)
print(x,y)
# Plot points
while circle.contains_point([x,y]): # see 1st figure
#for i in range(5): # uncomment this line (for) and comment above line (while) see 2nd figure
xnew = x**2 - y**2 + x0
ynew = 2 * x * y + y0
x = xnew
y = ynew
# Update current point
curr_pt = [x,y]
# Plot point (x,y)
plt.plot(x,'.c')
counter += 1
# Post-checks
print(circle.contains_point([x,y)
print(xnew,ynew) # note this check will output error with while loop as they will not be defined
# Print plot
plt.show()
输出:
# while loop (fig 1)
False
0
-1 0.5
-1 0.5
False
0
-1 0.5
-1 0.5
# for loop (fig 2)
False
0
-1 0.5
-1 0.5
False
5
-1 0.5
5.063203536206856 -4.162733919918537
5.063203536206856 -4.162733919918537
解决方法
更新:
利用这些答案中概述的方法:
- matplotlib.patches.Polygon.get_path().contains_point(x,y)
- Circle.get_path(),get_transform(),transform_path()
我修改后的代码现在可以正确识别点是否在 Circle
中,并根据 while
执行 contains_point()
循环。
修改后的代码:
# IMPORT LIBRARIES ####
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.path as mPath
import matplotlib.patches as mPatches
# CREATE CIRCLE ####
# Input circle radius
circleRadius = 2
# Create circle (arguments: centre,radius,face colour (fc),edge colour (ec),alpha)
circle = mPatches.Circle((0,0),radius = circleRadius,fc = 'm',alpha = 0.3)
# Get path and 2D affine transformation
pathC = circle.get_path()
transformC = circle.get_transform()
# Apply transformation to the path
transPath = transformC.transform_path(pathC)
# Get path of transformed circle
polygon = mPatches.PathPatch(transPath,alpha = 0.3)
# CREATE POINTS ####
# Input initial coordinates
x0 = -1
y0 = 0.5
# Assignment for new point equation
x = x0
y = y0
# Store
curr_pt = (x,y)
all_x = []
all_y = []
# Iterations required to escape circle
counter = 0
# Pre-check
cp1 = polygon.contains_point(curr_pt)
print("Pre-loop checks:")
print("Polygon contains point:",cp1)
print("Counter:",counter)
print("(x,y):",x,y)
print("")
# Create figure and single subplot
fig,ax = plt.subplots()
# Plot initial point (x,y)
ax.scatter(curr_pt[0],curr_pt[1],color = 'c')
# Plot new points
while polygon.contains_point(curr_pt):
xnew = x**2 - y**2 + x0
ynew = 2 * x * y + y0
x = xnew
y = ynew
# Update current point
curr_pt = (x,y)
# Update x,y lists for plotting lines
all_x.append(curr_pt[0])
all_y.append(curr_pt[1])
# Plot new point
ax.scatter(curr_pt[0],color = 'c')
# Increment counter
counter += 1
if counter == 50:
break
# Post-check
cp2 = polygon.contains_point(curr_pt)
print("Post-loop checks:")
print("Polygon contains point:",cp2)
print("Counter:",y)
print("(xnew,ynew):",xnew,ynew)
# CREATE PLOT ####
# Plot lines
ax.plot(all_x,all_y,color = 'c')
# Plot circle
ax.add_patch(polygon)
# Print plot
ax.set_aspect(1.0)
fig.savefig("Correct point in polygon (points and lines).png")
plt.show()
输出:
Pre-loop checks:
Polygon contains point: True
Counter: 0
(x,y): -1 0.5
Post-loop checks:
Polygon contains point: False
Counter: 4
(x,y): -2.6183929443359375 0.890380859375
(xnew,ynew): -2.6183929443359375 0.890380859375