python - 基于 0 和非 0 的向量创建 0 和 1 的矩阵

Question

给定一个向量 v=[0, 0, 0, 0, 2, 0, 0, 0, 2.5, 0, 0, 0]

我想创建一个矩阵，其中 num_rows = np.count_nonzero(v) 和 num_cols = len(v) 为 0 和 1，如下面的输出所示。我不清楚如何生成这样的矩阵。

输出:

[[ 0.  0.  0.  0.  1.  0.  0.  0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.  0.  0.  0.  1.  0.  0.  0.]]

Answer 1

试试这个:

m = np.zeros((np.count_nonzero(a), len(a)))
row_index = 0
for i in range(len(a)):
    if a[i] != 0:
        m[row_index][i] = 1
         row_index += 1