Question

我使用Python学习线性代数，我有两个词典：

v = {1: 1, 2: 8, 3: 0}

和

M = {(1, 2): 2, (3, 1): 1, (3, 3): 7, (2, 1): -1}

我想创建一个字典，添加所有元素，其中v中的键与M中元组的第一部分相同。这里我想要作为两个字典的答案。（我将展示我想要做的计算）

newDict = {1: 1*M[(1, 2)], 2: 8*M[(2, 1)], 3: 0*M[(3, 1)]+0*M[(3, 3)]

与：

相同

newDict = {1: 1*2, 2: 8*-1, 3: 0*1+3*7}

所以我得到了表格中的最终字典

newDict = {1:2, 2:-8, 3:0}

正如你所看到的，我想要与字典v中相同的键。我得到的最接近的是：

>>> [v[k]*M[r] for k in v for r in M if k == r[0]]
[2, -8]

我至少有正确的答案，但我不能让它发挥作用。我不知道从这里去哪里或者我是否在正确的轨道上。对不起，如果我的解释可能缺乏

Answer 1

因为您基于多个输入键的值，所以使用循环而不是理解。使用collections.defaultdict object也可以使逻辑更简单：

from collections import defaultdict

newDict = defaultdict(int)
for x, y in M:
    newDict[x] += M[x, y] * v.get(x, 0)

输出：

>>> from collections import defaultdict
>>> v = {1: 1, 2: 8, 3: 0}
>>> M = {(1, 2): 2, (3, 1): 1, (3, 3): 7, (2, 1): -1}
>>> newDict = defaultdict(int)
>>> for x, y in M:
...     newDict[x] += M[x, y] * v.get(x, 0)
... 
>>> newDict
defaultdict(<type 'int'>, {1: 2, 2: -8, 3: 0})

Answer 2

这个怎么样..

newD = {k : 0 for k in v}
for k in v:
    for r in M:
        if k == r[0]: newD[k] += v[k]*M[r]

Python：dict乘法

2 个答案: