https://keroseneheaterreview.blogspot.com/2023/06/numpy-stack-to-2d-array-select-by-index.html