https://travelebookdownload.blogspot.com/2022/07/numpy-stack-to-2d-array-select-by-index.html