Карта функции части Подсписка питон

Question

Я работаю в Python и у меня есть список списков (или может быть также NumPy массива), как это:

list = [[0, 0, -1], [1, 1, -2], [1, 2, -3], [2, 2, -4]] 

У меня есть функция, которая принимает для каждого вывести первые два элемента и дать результат. Предположим, что функция Я принимаю о том, сумма() так, что я хочу сделать, это

list1 = [[sum(0, 0), -1], [sum(1, 1), -2], [sum(1, 2), -3], [sum(2, 2), -4]] 

и выход

list1 = [[0, -1], [2, -2], [3, -3], [4, -4]] 

Есть ли способ это с map или numpy.vectorize? Я уже могу сделать это как цикл, но это слишком дорого для моих реальных данных.

Любая помощь, совет?

Answer 1

Может ли это быть

[(sum(x, y), z) for x, y, z in list] 

Или, если вы настаиваете на использовании map использования itertools.starmap:

starmap(lambda x, y, z: (x+y, z), list) 

?

Answer 2

Для вашего конкретного примера, вы можете использовать список понимание следующим образом:

my_list = [[0, 0, -1], [1, 1, -2], [1, 2, -3], [2, 2, -4]] 

answer = [[x + y, z] for x, y, z in my_list] 
print(answer) 

Выход

[[0, -1], [2, -2], [3, -3], [4, -4]] 

Чтобы применить более сложную общую функцию для каждого элемента в вашем списке, использование map

def func(sublist): 
    return [sum(sublist[:2]), sublist[-1]] 

my_list = [[0, 0, -1], [1, 1, -2], [1, 2, -3], [2, 2, -4]] 

answer = map(func, my_list) 

С помощью функции func вы можете настроить свою функцию по своему усмотрению.

Answer 3

В случае Numpy массива вы должны иметь самые быстрые результаты с помощью векторизации:

In [128]: arr = np.asarray(list) 
#Out[128]: 
#array([[ 0, 0, -1], 
#  [ 1, 1, -2], 
#  [ 1, 2, -3], 
#  [ 2, 2, -4]]) 

In [129]: np.vstack((arr[:,0]+arr[:,1], arr[:,2])).T 
#Out[129]: 
#array([[ 0, -1], 
#  [ 2, -2], 
#  [ 3, -3], 
#  [ 4, -4]])