python - 组合功能

Question

我需要一个函数的帮助，该函数输出一个字典，其中包含 R1 和 R2 之间的组合，它的输出表示 R2 ◦ R1

到目前为止，我已经

def composition(R1,R2):
  d = {i:R2[R1[i]] for i in R1}
  print(d)

此功能适用于输入

R1 = {1:'a',2:'b',3:'c'}
R2 = {'a':'A','b':'B','c':'C'}
output: {1: 'A', 2: 'B', 3: 'C'}

我需要工作的是

R1 = {1 : [2, 3], 2 : [3], 3 : [4], 4 : [1, 3]}
R2 = {1 : [3], 2 : [1, 4], 3 : [2], 4 : [1, 2]}
output should be: R2 ◦ R1 = {1 : [1, 2, 4], 2 : [2], 3 : [1, 2], 4 : [2, 3]}

我得到的是不可散列的类型列表

任何帮助都会很棒，谢谢。

Answer 1

单独使用列表理解在Python中编写有点复杂(也许通过使用functools.reduce折叠子列表可以提高可读性)，但它应该是这样的:

def composition(R1, R2):
  d = {
      i: [
          val
          for sublist in [R2[j] for j in R1[i]]
          for val in sublist
      ]
      for i in R1
  }

  print(d)