2018-11-14
阅读量:
940
Python里怎么指定输出
在进行大量运算时, 有时候指定一个用于存放运算结果的数组是非
常有用的。 不同于创建临时数组, 你可以用这个特性将计算结果直
接写入到你期望的存储位置。 所有的通用函数都可以通过 out 参数
来指定计算结果的存放位置:
In[24]: x = np.arange(5)
y = np.empty(5)
np.multiply(x, 10, out=y)
print(y)
[ 0. 10. 20. 30. 40.]
这个特性也可以被用作数组视图, 例如可以将计算结果写入指定数
组的每隔一个元素的位置:
In[25]: y = np.zeros(10)
np.power(2, x, out=y[::2])
print(y)
[ 1. 0. 2. 0. 4. 0. 8. 0. 16. 0.]
如果这里写的是 y[::2] = 2 ** x, 那么结果将是创建一个临时
数组, 该数组存放的是 2 ** x 的结果, 并且接下来会将这些值复
制到 y 数组中。 对于上述例子中比较小的计算量来说, 这两种方式
的差别并不大。 但是对于较大的数组, 通过慎重使用 out 参数将能
够有效节约内存。






评论(0)


暂无数据
推荐帖子
0条评论
0条评论
0条评论