shapely geometry - how to preserve refrence to original feature

Question

What is preferred method to save reference from geometry to original feature when working with shapely geometries in python?

Ie. After creating STRtree and getting some result from it, there is no direct way to access feature f from which this geometry was created.

from shapely.geometry import shape
from shapely.strtree import STRtree
...
tree = STRtree([shape(f['geometry']) for f in features])

I guess there is something short easy and cheap as this is often used but can't find it or guess what it is? Dict comes first to mind but geometries are unhashable so?

Answer 1

The shapely STRtree documentation explicitly mentions how to deal with this in a note, I guess you missed it or even checking it might not have come to mind? https://shapely.readthedocs.io/en/stable/manual.html#strtree.STRtree

To get the original indexes of the query results, create an auxiliary dictionary. But use the geometry ids as keys since the shapely geometries themselves are not hashable.

index_by_id = dict((id(pt), i) for i, pt in enumerate(points))
[(index_by_id[id(pt)], pt.wkt) for pt in tree.query(Point(2,2).buffer(1.0))]

[(1, 'POINT (1 1)'), (2, 'POINT (2 2)'), (3, 'POINT (3 3)')]

Answer 2

Until nice answer found I have used this ugly hack to save data which I need:

def shape_save_props(feat):
    s = shape(feat['geometry'])
    s.properties = feat['properties']
    return s