python - 检查字典中键的值并仅使用 Python 为键构造矩阵

Question

我有一本字典，如下所述。

mydictionary = {colours: [red, pink, blue, green, yellow], animals: [cat, rat, dog, goat], 
vehicles: [car, jeep, van, bus, lorry]}

对于键的每个值，我还有一个矩阵，如下例所示(如果该值在我阅读的文章中，我将其设为 1，否则设为 0)。

        red, pink, blue, green, yellow, cat, rat, ... bus, lorry
Essay1:  1     0     0    1       1      0    0        1     0
Essay2:  0     1     0    0       1      0    1        0     0

因此，这看起来像一个列表列表。

[[1,0,0,1,1,0,0,1,0], [0,1,0,0,1,0,1,0,0]]

现在，通过使用“mydictionary”，我想按如下方式转换上述矩阵(也就是说，如果键的一个或多个值是 1，我将键标记为 1，否则标记为 0)。

      Colours, Animals, Vehicles
Essay1: 1         0        1
Essay2: 1         1        0

上面提到的矩阵可以写成如下列表的列表。

[[1,0,1], [1,1,0]]

我是 pandas 的新手，因此，我很想知道这是否可以使用 pandas 数据帧来完成。

Answer 1

m = {v: k for k, l in mydictionary.items() for v in l}
df.groupby(df.columns.map(m.get), 1).sum().clip(0, 1)

        animals  colours  vehicles
Essay1        0        1         1
Essay2        1        1         0