在Python中计算数组的余弦值

时间:2014-01-10 11:51:16

标签: python arrays math numpy cosine

我有一个名为a的1242个数字的数组。我需要获取Python中所有数字的余弦值。

当我使用:cos_ra = math.cos(a)时,我收到错误消息:

  

TypeError:只能将length-1数组转换为Python标量

我该如何解决这个问题?

提前致谢

5 个答案:

答案 0 :(得分:5)

问题是你在这里使用numpy.math.cos,它希望你传递一个标量。如果要将numpy.cos应用于可迭代,请使用cos

In [30]: import numpy as np

In [31]: np.cos(np.array([1, 2, 3]))                                                             
Out[31]: array([ 0.54030231, -0.41614684, -0.9899925 ])

错误:

In [32]: np.math.cos(np.array([1, 2, 3]))                                                        
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-32-8ce0f3c0df04> in <module>()
----> 1 np.math.cos(np.array([1, 2, 3]))

TypeError: only length-1 arrays can be converted to Python scalars

答案 1 :(得分:4)

使用numpy

In [178]: from numpy import *

In [179]: a=range(1242)

In [180]: b=np.cos(a)

In [181]: b
Out[181]: 
array([ 1.        ,  0.54030231, -0.41614684, ...,  0.35068442,
       -0.59855667, -0.99748752])

此外,numpy数组操作非常快:

In [182]: %timeit b=np.cos(a)  #numpy is the fastest
10000 loops, best of 3: 165 us per loop

In [183]: %timeit cos_ra = [math.cos(i) for i in a]
1000 loops, best of 3: 225 us per loop

In [184]: %timeit map(math.cos, a)
10000 loops, best of 3: 173 us per loop

答案 2 :(得分:3)

问题是math.cos期望在尝试传递列表时获取数字作为参数。 您需要在每个列表元素上调用math.cos

尝试使用map

map(math.cos, a)

答案 3 :(得分:1)

math.cos()只能在单个值上调用,而不能在列表中调用。

另一种选择,使用列表理解:

cos_ra = [math.cos(i) for i in a]

答案 4 :(得分:1)

张小晨的回答很容易。

np.cos(np.arange(start, end, step))

希望这会有所帮助!