python numpy圆函数奇怪的错误

时间:2016-04-03 00:49:23

标签: python numpy pandas

我有来自python for book的以下代码。但来自numpy的圆函数导致了一个错误,如"返回(小数,输出) TypeError:round()最多需要2个参数(给定3个)"

谁知道我做错了什么?

import numpy as np
import pandas as pd
import pandas.io.data as web

sp500 = web.DataReader('^GSPC', data_source='yahoo',
                            start='1/1/2000', end='4/14/2014')
sp500.info()
sp500['Close'].plot(grid=True, figsize=(8, 5))

sp500['42d'] = np.round(pd.rolling_mean(sp500['Close'], window=42), 2)

1 个答案:

答案 0 :(得分:3)

根据错误消息,似乎在numpy 1.11.0中,舍入函数如下所示:

try:
    round = a.round
except AttributeError:
    return _wrapit(a, 'round', decimals, out)
return round(decimals, out)

看起来pandas.Series.round只接受两个参数(selfprecision),但numpy传递了另一个参数out。据推测,这可能是pandasnumpy中的错误或API更改。

我可以看到两种简单的解决方法。第一种是直接使用Series.round()函数:

sp500['42d'] = pd.rolling_mean(sp500['Close'], window=42).round(2)

另一种选择是将numpy.round函数应用于基础numpy数组:

sp500['42d'] = np.round(pd.rolling_mean(sp500['Close'], window=42).values, 2)

修改:看起来这是一个已知问题。请参阅the pandas github tracker, issue #12644