Python:使用列表理解生成对称数组

Question

我正在尝试生成矩阵 A == [f(i,j) for i,j in range(0,n)] .矩阵是对称的 (f(i,j) == f(j,i))并且对角线元素为零 (f(i,i) == 0) .

问题:是否有可能产生一个列表理解来生成这个矩阵，但只调用函数 f(i,j)对于 i < j ？

不是我的问题:以其他方式生成对称矩阵。

---

可能的解决方案:调用f通过辅助功能 g保存 f 的值在额外的存储中或返回存储的值。

是否可以解决避免额外存储的问题？ 我不确定这个额外的障碍是否意味着有必要引用列表理解本身(我读到的在 Python 中不存在)，希望我错过了一些其他技巧。

Answer 1

假设您确实需要完整的矩阵，尽管它是对称的，您可以这样做。

首先创建矩阵A。 (我知道，这意味着您不是在列表理解中生成它。)这里我定义了一个“虚拟”函数 f 并假设 n 的值为 3。

>>> n=3
>>> subs=((i,j) for i in range(n) for j in range(3) if i<=j)
>>> for i,j in subs:
...     i,j
...     
(0, 0)
(0, 1)
(0, 2)
(1, 1)
(1, 2)
(2, 2)
>>> def f(i,j):
...     return
... 
>>> def g(i,j):
...     if i==j:
...         A[i,i] = 0
...     else:
...         val=f(i,j)
...         A[i,j]=val
...         A[j,i]=val

如果我没听错的话，不需要额外的存储空间。在练习 subs 的列表理解中调用 g。