我有以下课程。
class PlotOnMap:
def __init__(self, lat_lon_a, lat_lon_b, optimal, f_map=None):
self.lat_lon_a = lat_lon_a
self.lat_lon_b = lat_lon_b
self.optimal = optimal
self.f_map = f_map
def create_map(self):
"""
This function creates a map with folium, centered with the location of the three given points.
"""
self.f_map = folium.Map([midlat, midlon], zoom_start=12)
def plot_gdt_and_triangulation(self):
"""
This function plots the locations given by the users with markers and create an heat map of the triangulation.
"""
# plot the gdt's locations.
folium.Marker(
location=gdt1,
popup='GDT 1',
icon=folium.Icon()
).add_to(self.f_map)
folium.Marker(
location=gdt2,
popup='GDT 2',
icon=folium.Icon()
).add_to(self.f_map)
folium.Marker(
location=uav,
popup='Target Area.',
icon=folium.Icon()
).add_to(self.f_map)
# plot the triangulation (regular and optimal) area
plugins.HeatMap(data=self.lat_lon_a,
).add_to(self.f_map)
plugins.HeatMap(data=self.lat_lon_b,
).add_to(self.f_map)
plugins.HeatMap(data=self.optimal,
gradient={0.65: 'lime'}
).add_to(self.f_map)
class Plot2ndPosition(PlotOnMap):
def __init__(self, lat_lon_a, lat_lon_b, optimal):
self.plot_on_map = super().__init__(lat_lon_a, lat_lon_b, optimal)
def plot_map(self):
return self.plot_on_map.create_map()
def plot_locations(self):
return self.plot_on_map.plot_gdt_and_triangulation()
唯一的区别是,在Plot2ndPosition类中,在第二种方法中,我不想绘制gdt2和热图,我只想绘制gdt1和uav(第一个和最后一个标记)。
我考虑了一些事情,将每个变量赋值给一个变量,并在继承方法时以某种方式省略它。
def plot_gdt_and_triangulation(self):
"""
This function plots the locations given by the users with markers and create an heat map of the triangulation.
"""
# plot the gdt's locations.
one = folium.Marker(
location=gdt1,
popup='GDT 1',
icon=folium.Icon()
)
one.add_to(self.f_map)
two = folium.Marker(
location=gdt2,
popup='GDT 2',
icon=folium.Icon()
)
two.add_to(self.f_map)
three = folium.Marker(
location=uav,
popup='Target Area.',
icon=folium.Icon()
)
three.add_to(self.f_map)
# plot the triangulation (regular and optimal) area
trian_plot1 = plugins.HeatMap(data=self.lat_lon_a,
).add_to(self.f_map)
trian_plot2 = plugins.HeatMap(data=self.lat_lon_b,
).add_to(self.f_map)
trian_plot3 = plugins.HeatMap(data=self.optimal,
gradient={0.65: 'lime'}
).add_to(self.f_map)