Numpy slice. Массивы numpy. numpy reshape. numpy axis. numpy документация.