Карта лямбда в Python 3

Я пытаюсь использовать пример сюжета в Python 3, но получаю синтаксическую ошибку в этой строке:

return map(lambda (x, y, an): (x, y), cornersWithAngles)

Я уже читал, что использование скобок для распаковки аргументов в лямбде недопустимо в Python 3, но я не знаю, как именно настроить свой код для решения этой проблемы.

Вот полный код (ошибка в строке 16):

import plotly.plotly as py
import plotly.graph_objs as go
from plotly.tools import FigureFactory as FF

import scipy

def PolygonSort(corners):
    n = len(corners)
    cx = float(sum(x for x, y in corners)) / n
    cy = float(sum(y for x, y in corners)) / n
    cornersWithAngles = []
    for x, y in corners:
        an = (np.arctan2(y - cy, x - cx) + 2.0 * np.pi) % (2.0 * np.pi)
        cornersWithAngles.append((x, y, an))
    cornersWithAngles.sort(key = lambda tup: tup[2])
    return map(lambda (x, y, an): (x, y), cornersWithAngles)

def PolygonArea(corners):
    n = len(corners)
    area = 0.0
    for i in range(n):
        j = (i + 1) % n
        area += corners[i][0] * corners[j][1]
        area -= corners[j][0] * corners[i][1]
    area = abs(area) / 2.0
    return area

corners = [(0, 0), (3, 0), (2, 10), (3, 4), (1, 5.5)]
corners_sorted = PolygonSort(corners)
area = PolygonArea(corners_sorted)

x = [corner[0] for corner in corners_sorted]
y = [corner[1] for corner in corners_sorted]

annotation = go.Annotation(
    x=5.5,
    y=8.0,
    text='The area of the polygon is approximately %s' % (area),
    showarrow=False
)

trace1 = go.Scatter(
    x=x,
    y=y,
    mode='markers',
    fill='tozeroy',
)

layout = go.Layout(
    annotations=[annotation],
    xaxis=dict(
        range=[-1, 9]
    ),
    yaxis=dict(
        range=[-1, 12]
    )
)

trace_data = [trace1]
fig = go.Figure(data=trace_data, layout=layout)

py.iplot(fig, filename='polygon-area')

2 ответа

Просто используйте x как кортеж, может быть, это может помочь:

map(lambda x: (x[0], x[1]), cornerWithEdges)

Что вы пытаетесь сделать (распаковка параметров в lambda) работал в Python 2, но больше не работает в Python 3.

Python 2.7.15rc1 (default, Nov 12 2018, 14:31:15) 
>>> lst = [(1,2,3), (4,5,6), (7,8,9)]
>>> map(lambda (a,b,c): (a,b), lst)
[(1, 2), (4, 5), (7, 8)]

Python 3.6.7 (default, Oct 22 2018, 11:32:17) 
>>> lst = [(1,2,3), (4,5,6), (7,8,9)]
>>> map(lambda (a,b,c): (a,b), lst)
  File "<stdin>", line 1
    map(lambda (a,b,c): (a,b), lst)
               ^
SyntaxError: invalid syntax

Однако есть несколько вещей, которые вы можете сделать.

>>> list(map(lambda t: t[:2], lst))
[(1, 2), (4, 5), (7, 8)]
>>> [(a,b) for a, b, c in lst]
[(1, 2), (4, 5), (7, 8)]
>>> from itertools import starmap
>>> list(starmap(lambda a, b, c: (a, b), lst))
[(1, 2), (4, 5), (7, 8)]

Лично я бы пошел со списком-пониманием или генератором-выражением. Также помните, что в Python 3 map это генератор, а не список, поэтому при использовании map вам, возможно, придется обернуть результат в listв зависимости от того, что вы хотите с ним делать.

Другие вопросы по тегам