列出一級“拆包”

Question

我有一個返回 Sympy 點的 function：

result = [Point3D(500, 500, 10), Point3D(-500, 500, 10), Point3D(-500, -500, 10), Point3D(500, -500, 10)]

但我需要獲取點的值，而不是點實例，所以我這樣做：

a, b, c, d = [], [], [], []
for i in range(3):
    a.append(result[0][i])
for i in range(3):
    b.append(result[1][i])
for i in range(3):
    c.append(result[2][i])
for i in range(3):
    d.append(result[3][i])

# And then pack it:
result = [tuple(a), tuple(b), tuple(c), tuple(d)]

我知道這是獲得所需輸入的一種可怕方式，即：

result = [(500, 500, 10), (-500, 500, 10), (-500, -500, 10), (500, -500, 10)]

我怎樣才能以更好的方式做到這一點？

Answer 1

使用列表推導：

result = [tuple([p[0], p[1], p[2]]) for p in result]

我剛剛檢查了Point3D的文檔，看來您可以有一個更優雅的解決方案：

result = [tuple([p.x, p.y, p.z]) for p in result]

或者正如@OscarBenjamin 在評論中指出的那樣，這可能是最簡單的解決方案，您可以使用：

result = [p.args for p in result]

Answer 2

小提示：（由我）向 sympy 的點 class 添加了一個新方法，該方法返回該點的所有坐標：

result = [p.coordinates for p in result]