Avviso Pyrep Getitem per funzioni con array

Question

Ricevo avvisi di ispezione del codice da PyCharm. Capisco la logica, ma non sono chiaro sul modo appropriato per risolverlo. Dire Ho la seguente funzione di esempio:

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

ottengo 2 avvisi:

>> 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) 

Lo scopo della funzione è naturalmente per analizzare una matrice di NumPy xdata. Ma PyCharm non lo sa, quindi senza ulteriori indicazioni si assume che xdata (e quindi anche ydata) sia un numero intero.

Qual è il modo appropriato per rispondere a questo avviso? Devo notare che l'aggiunta di una riga di controllo del tipo risolverà l'avviso. È la soluzione ottimale? Per esempio:

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

Infine, aggiungendo Sfinge informazioni docstring non sembra avere alcun effetto sulle avvertenze. (l'avviso vede ancora 'int' quando xdata è specificato come str). Anche l'iterazione di y traduce direttamente nella seguente errore:

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

Answer 1

PyCharm ha type hinting caratteristiche che possono risultare utili.

Per esempio, in questo caso, il codice seguente rende gli errori vanno via:

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