带有数组的函数的PyCharm getitem警告

时间:2015-03-31 16:48:14

标签: python numpy iterator pycharm

我从PyCharm收到代码检查警告。我理解逻辑,但我不清楚修复它的适当方法。假设我有以下示例函数:

def get_ydata(xdata):
    ydata = xdata ** 2
    for i in range(len(ydata)):
        print ydata[i]
return ydata

我得到2个警告:

>> Expected type 'Sized', got 'int' instead (at line 3)
>> Class 'int' does not define '__getitem__', so the '[]' operator cannot be used on its instances (at line 4)

该函数的目的当然是解析一个numpy数组的xdata。但PyCharm并不知道,所以没有任何进一步的指示假设xdata(因此也是ydata)是一个整数。

解决此警告的适当方法是什么?我应该注意,添加类型检查行将修复警告。这是最佳解决方案吗?例如:

if not type(ydata) is np.ndarray:
    ydata = np.array(ydata)

最后,添加Sphinx文档字符串信息似乎对警告没有任何影响。 (当xdata指定为str时,警告仍会看到'int')。同样迭代y直接导致以下错误:

for y in ydata:
...
>> Expected 'collections.Iterable', got 'int' instead

2 个答案:

答案 0 :(得分:4)

Pycharm具有可能有用的type hinting功能。

例如,在这种情况下,以下代码会使错误消失:

import numpy as np

def get_ydata(xdata):
    ydata = xdata ** 2  # type: np.ndarray
    for i in range(len(ydata)):
        print(ydata[i])
    return ydata

最近的python版本还包括对type annotations

的支持
import numpy as np
def get_ydata(xdata: np.ndarray):
    ...

答案 1 :(得分:0)

TL; DR 使用list()投射

还是很晚

我在其他一些代码上也遇到类似的问题。

我可以通过类似于

的方法解决它
def get_ydata(xdata):
    ydata = list(xdata ** 2)
    for i in range(len(ydata)):
        print ydata[i]
    return ydata