热线电话:13121318867

登录
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.0000
4
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子