任务描述
本关任务:给定两个不同形状的数组,求出他们的和。
相关知识
广播(Broadcast)是 numpy 对不同形状(shape)的数组,进行数值计算的方式。 对数组的算术运算通常在相应的元素上进行,当运算中的 2 个数组的形状不同时,numpy 将自动触发广播机制。
如图所示:
广播的规则
- 让所有输入数组都向其中形状最长的数组看齐,形状中不足的部分都通过在前面加 1 补齐;
- 输出数组的形状是输入数组形状的各个维度上的最大值;
- 如果输入数组的某个维度和输出数组的对应维度的长度相同,或者其长度为 1 时,这个数组能够用来计算,否则出错;
- 当输入数组的某个维度的长度为 1 时,沿着此维度运算,都用此维度上的第一组值。
编程要求
首先用arange()生成一个数组,然后用reshape()方法,将数组切换成4x3的形状,最后再与basearray相加,输出它们的和。
请按照编程要求,补全右侧编辑器Begin-End区间的代码。
测试说明
平台会对你编写的代码进行测试:
测试输入:12;
预期输出:
[[ 1 3 5]
[ 4 6 8]
[ 7 9 11]
[10 12 14]]
开始你的任务吧,祝你成功!*
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))