in Matlab, è possibile assegnare un valore a più sezioni della stessa lista:assegnare il valore a più sezioni in NumPy
>> a = 1:10
a =
1 2 3 4 5 6 7 8 9 10
>> a([1:3,7:9]) = 10
a =
10 10 10 4 5 6 10 10 10 10
Come si può fare questo in Python con un allineamento numpy?
>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> a[1:3,7:9] = 10
IndexError: too many indices
a = np.arange(10)
a[[range(3)+range(6,9)]] = 10
#or a[[0,1,2,6,7,8]] = 10
print a
che dovrebbe funzionare penso ... Non so che il suo proprio quello che vuoi se
Si potrebbe anche considerare l'utilizzo di np.r_:
http://docs.scipy.org/doc/numpy/reference/generated/numpy.r_.html
ii = np.r_[0:3,7:10]
a[ii] = 10
In [11]: a
Out[11]: array([ 10, 10, 10, 3, 4, 5, 6, 10, 10, 10])
Da http://docs.scipy.org/doc/numpy/user/basics.indexing.html (Sezione "Matrici di indici"). Si noti che gli indici per le sezioni desiderate dovrebbero essere contenuti in 'np.array()'.
>>> x = np.arange(10,1,-1)
>>> x
array([10, 9, 8, 7, 6, 5, 4, 3, 2])
>>> x[np.array([3, 3, 1, 8])]
array([7, 7, 9, 2])
Qualsiasi motivo per cui si potrebbe desiderare di farlo in una dichiarazione invece di farlo in due dichiarazioni? –
@SukritKalra Non posso parlare per l'autore della domanda, ma un caso esemplificativo è quando si cerca di formare una lista di comprensione da un tale multislice (che è quello che sto cercando di fare e perché sono arrivato a questa domanda) – Pyderman