将 1D numpy 数组的值转换为 2D numpy 数组中的列号 [重复]

Question

我需要以下逻辑：

• 输出数组的行数与 输入数组中的元素
• 输入数组的值表示输出数组的列数（列数等于输入数组的最大数）
• 如果输入数组包含特定组合，则输出数组的值应为 1，否则为 0

例如，我想转换一个 numpy 数组，例如

[2, 1, 1, 0]

放入一个看起来像这样的数组

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

我唯一能想到的就是用 for 循环填充一个空数组。还有更多“numpy-ish”的方法吗？

Answer 1

使用这个：

before = [2, 1, 1, 0]

n_rows = len(before)
n_columns = max(before) + 1

output = np.zeros((n_rows, n_columns))

#loop constructor
for idx,i in enumerate(before): output[idx,i] = 1

array([[0., 0., 1.],
       [0., 1., 0.],
       [0., 1., 0.],
       [1., 0., 0.]])