a_adli/varennes_gis_oo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add centroid fields & Convert to a DF

Browse Source
This commit is contained in:
Alireza Adli 2024-08-23 16:41:08 -04:00
parent 042913acdc
commit 7242159aaa
1 changed files with 54 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

61
varennes_single_processes/remove_nrcan_duplicates.py
View File

@ -7,24 +7,71 @@ Project Developer: Alireza Adli alireza.adli@concordia.ca
# You need to clone mtl_gis_oo project and
# add it as a dependency of this new project
from scrub_layer_class import *
import pandas as pd
# Change the paths by the location of your QGIS installation and datalayers
qgis_path = 'C:/Program Files/QGIS 3.34.1/apps/qgis'
varennes_nrcan_extra_polygons = \
'C:/Users/a_adli/PycharmProjects/varennes_gis_oo/' \
'data/initial_data/endeavor/nrcan_centroid_main/nrcan_centroid_main.shp'
'data/initial_data/endeavor/nrcan_without_centroids/auto_building_2.shp'
# first we duplicate the layer to preserve the main data layer.
# First we duplicate the layer to preserve the main data layer.
duplicated = \
'C:/Users/a_adli/PycharmProjects/varennes_gis_oo/' \
'data/initial_data/endeavor/nrcan_centroids_1/auto_building_2.shp'
# First, the layer will be duplicated as some records are going to be removed
'data/initial_data/endeavor/nrcan_centroids/nrcan_centroids.shp'
varennes_nrcan = ScrubLayer(
qgis_path, varennes_nrcan_extra_polygons, 'NRCan Varennes')
varennes_nrcan.duplicate_layer(duplicated)
varennes_nrcan_centroids = ScrubLayer(
qgis_path, duplicated, 'NRCan Varennes with Coordinates')
# Then we add coordinates to each polygon so we can remove the
# very similar polygons based on coordinates.
varennes_nrcan_centroids.layer.startEditing()
# Add new fields for the centroid coordinates
varennes_nrcan_centroids.layer.dataProvider().\
addAttributes([QgsField("centroid_x", QVariant.Double),
QgsField("centroid_y", QVariant.Double)])
varennes_nrcan_centroids.layer.updateFields()
centroid_x_index = varennes_nrcan_centroids.\
layer.fields().indexFromName("centroid_x")
centroid_y_index = varennes_nrcan_centroids.\
layer.fields().indexFromName("centroid_y")
for feature in varennes_nrcan_centroids.layer.getFeatures():
centroid = feature.geometry().centroid().asPoint()
feature.setAttribute(centroid_x_index, centroid.x())
feature.setAttribute(centroid_y_index, centroid.y())
varennes_nrcan_centroids.layer.updateFeature(feature)
# Commit the changes for adding centroids fields
varennes_nrcan_centroids.layer.commitChanges()
# Pandas is a better option to compare polygons and remove the duplicates
# so we make a dataframe. We just transfer the necessary fields
# to the dataframe.
field_names = ['feature_id', 'centroid_x', 'centroid_y']
# Get the indices of the specified fields
field_indices = [varennes_nrcan_centroids.layer.fields().indexOf(field)
for field in field_names]
# Extract the attribute values and store them in a list of dictionaries
data = []
for feature in varennes_nrcan_centroids.layer.getFeatures():
attributes = [feature.attributes()[index] for index in field_indices]
data.append(dict(zip(field_names, attributes)))
# Create a DataFrame from the list of dictionaries
varennes_nrcan_centroids_df = pd.DataFrame(data)
varennes_nrcan_centroids_df['ID'] = range(len(varennes_nrcan_centroids_df))
varennes_nrcan_duplicate = varennes_nrcan.duplicate_layer(
duplicated, 'NRCan duplicated')