头歌python Numpy进阶

第1关：Numpy广播

任务描述
本关任务：给定两个不同形状的数组，求出他们的和。

编程要求
首先用arange()生成一个数组，然后用reshape()方法，将数组切换成4x3的形状，最后再与basearray相加，输出它们的和。

请按照编程要求，补全右侧编辑器Begin-End区间的代码。

import numpy as np

basearray = eval(input())
# *********** Begin ************ #
arr = np.arange(12).reshape(4,3)


# ************ End ************* #
print("{} + \n{}".format(arr,basearray))
print("结果为：\n{}".format(basearray+arr))

第2关：Numpy高级索引

任务描述
本关任务：给定一个二维数组，请以整数数组索引、布尔索引、花式索引三种方式，来获取我们需要的数组元素。

编程要求
首先，利用花式索引获取arr数组第line行（至少两行）的数组a；然后，利用整数数组索引获取数组四个角（按行优先的顺序获取）的元素得到数组b；最后，再利用布尔索引获取大于10的元素得到数组c，并输出c。
请按照编程要求，补全右侧编辑器Begin-End 区间的代码。

import numpy as np

'''
arr为一个ndarray类型的数组,line为花式索引的索引数组
'''
def advanced_index(arr,line):
    # ********** Begin *********** #
    a = arr[line]
    row,colum = a.shape
    b = a[[0,0,row-1,row-1],[0,colum-1,0,colum-1]]
    c = b[b>10]
    # *********** End ************ #
    return c

def main():
    line = list(map(lambda x:int(x),input().split(",")))
    arr = np.arange(24).reshape(6, 4)
    print(advanced_index(arr,line))

if __name__ == '__main__':
    main()

第3关：Numpy迭代数组

任务描述
本关任务：利用本关相关知识，将一个ndarray类型的数组，顺时针旋转90度后输出。

编程要求
根据提示，在右侧编辑器Begin-End区间补充代码，将一个ndarray类型的数组顺时针旋转90度后输出。
提示：

建议使用外部循环；
np.vstack()可以将两个数组竖直堆叠成为新的数组。

import numpy as np

'''传入的参数为ndarray类型数组'''
def rotate_array(arr):
    # ******* Begin ******* #
    new_array = []
    for i in range(arr.shape[1]):
        new_row = arr[::-1,i]
        new_array.append(new_row)
    result = np.vstack(new_array)
    # ******* End ******** #
    return result

def main():
    arr = eval(input())
    print(rotate_array(arr))

if __name__ == '__main__':
    main()