Карта лямбда в 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
в зависимости от того, что вы хотите с ним делать.