Pythonic方法获取矩阵的某些行

I was thinking about a code that I wrote a few years ago in Python, at some point it had to get just some elements, by index, of a list of lists.I was thinking about a code that I wrote a few