Как передать кусок в функцию по ссылке

Question

Если у меня есть

a = [1, 2, 3] 

def foo (arr): 
    for i in len (arr): arr [i] += 1 

def bar (arr): 
    foo (arr[:2]) 

bar (a) 
print (a) 

Я хочу, чтобы результат как

>>> [2, 3, 3 ] 

Как я могу идти об этом?

Мотивация: мне нужна очередь приоритетов, где я могу заморозить последние N элементов, т.е. только пройти queue[:N] до heapq.heappush(). Но каждый раз, когда я передаю фрагмент, к нему или к какой-либо функции вообще, он отправляет копию фрагмента, а не фактический список функции, и поэтому мой список остается неизменным в конце.

Answer 1

Чтобы быть честным, вместо того, чтобы идти на ломтик, я бы просто передал индексы;

a = [1, 2, 3] 
def foo(array, start, stop, jmp= 1): 
    for idx in range(start, stop + 1, jmp): 
     array[idx] += 1 

def bar(array): 
    foo(array, 1, 2) 
bar(a) 
print(a) 
[1, 3, 4] 

Answer 2

Используйте список понимание и обновить первоначальный список, используя полный назначение ломтика с [:]:

def foo(arr): 
    arr[:] = [x+1 for x in arr] 

Trial:

>>> a = [1, 2, 3] 
>>> def foo(arr): 
... arr[:] = [x+1 for x in arr] 
... 
>>> foo(a) 
>>> a 
[2, 3, 4] 

Answer 3

нарезка список будет создать новый список с нарезанным содержимым, поэтому arr[:2] теряет ссылку на оригинал a.

Помимо этого, итерация, как и вы, не изменит список вообще, он просто меняет item и игнорирует значение.

Если вы хотите изменить определенные части списка, вы должны нести с собой ссылку на исходный список. Например, держать arr, перебрать кусочек его с enumerate(arr[:2]), а затем изменить arr:

a = [1, 2, 3] 

def foo(arr): 
    for i, item in enumerate(arr[:2]): 
     arr[i] = item + 1 

def bar(arr): 
    foo(arr) 

bar(a) 
print(a) 

печать дает теперь [2, 3, 3], удаляя кусок в enumerate результатов в [2, 3, 4]. Конечно, здесь нецелесообразно, вы можете сбросить его и просто оставить foo и сразу позвонить по телефону foo(a).

Answer 4

Это будет работать, если a - массив numpy. По умолчанию numpy фрагменты относятся к тому же блоку памяти, что и исходный массив.

import numpy as np 
a = np.array([1, 2, 3]) 

def foo(arr): 
    for i in range(len(arr)): arr[i] += 1 
    # or just arr += 1 

def bar(arr): 
    foo(arr[:2]) 

bar(a) 
print(a) 
# [2, 3, 3 ]