1.6 对多个元素同时应用某个操作
问题描述
对一个数组中的多个元素同时应用某个函数。
解决方案
使用NumPy的vecotrize:
讨论
NumPy的vectorize类将一个函数转换成另一个函数,这个函数能把某个操作应用在数组的全部元素或一个切片上。值得注意的是,vectorize本质上是在对所有元素循环执行某个操作,所以并不会提升性能。此外,使用NumPy的数组,我们可以对两个维度不同的数组执行操作(这是一种叫作广播的方法)。举个例子,对于上述问题使用广播(broadcasting)后,我们的解决方案更简单:
本书评论