数组索引。 中间列的打印元素

Question

hackerank array indexing问题描述如下 编写 function array_index 接受三个数字 n,n_row,n_col 并执行下面给出的数组操作

1. 创建一个形状为 (n_row,n_col) 的数组 x，具有前 n 个自然数。 2.打印最后一行的元素
2. 打印中间列的元素。 4.打印元素，重叠前两行后三列我的python代码如下

---

import numpy as np
def array_index(n, n_row, n_col):
    
        x = np.arange(n).reshape(n_row, n_col)
        print(x[n_col])
    
if __name__ == '__main__':
        
        n = int(input())
        n_row = int(input())
        n_col = int(input())
        array_index(n, n_row, n_col)

有人可以帮我完成这个问题的解决方案 PS：附上问题图片以供参考在此处输入图片描述

Answer 1

import numpy as np
x=np.arange(n).reshape(n_row,n_col)
print(x[-1])
y=int(n_col/2)
print(x[:,y])
print(x[0:2,-3:])