python - 二值化数组

Question

我有一个表示星期几的列向量

[1,2,2,3,4]

我需要对这个向量进行二值化，因为原始向量中的每一项都必须转换为一个向量，其中数字表示索引需要为 1，其余必须为 0。

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

Answer 1

通过用零组成二进制列表来做到这一点，除了在列表理解中的给定位置，这给出了一个很好的单行:

w=[1,2,2,3,4]
m = [[0]*(pos)+[1]+[0]*(9-pos-1) for pos in w]

结果:

m = [[0, 1, 0, 0, 0, 0, 0, 0, 0],
     [0, 0, 1, 0, 0, 0, 0, 0, 0],
     [0, 0, 1, 0, 0, 0, 0, 0, 0],
     [0, 0, 0, 1, 0, 0, 0, 0, 0],
     [0, 0, 0, 0, 1, 0, 0, 0, 0]]