I do a lot of data stuff, mostly in Python/AWS/Terraform
Keeping this canned answer handy:
Yes, and it isn't too hard! Please take the tour, read what's on-topic here, How to Ask, and the question checklist, and provide a minimal reproducible example. "Implement this feature for me" is off-topic for this site. You have to make an honest attempt, and then ask a specific question about your algorithm or technique. Welcome to Stack Overflow!