numpy-基础练习

输入整数n，创建一个3*4的二维数组，数组元素从n开始，依次递增。
输出第1行，第1，3列的元素
eg:n=1
[ 6 8]

import numpy as np

n = int(input(""))
arr = np.arange(n, n+12).reshape(3, 4)
print(arr)
print(arr[0][[0,2]])
print(arr[:,[0,2]][0])