代码之家 › 专栏 › 技术社区 › user1518183

NumPy:将一个热编码推广到k-hot编码

numpy

user1518183 · 技术社区 · 7 年前

我正在使用此代码对值进行热编码:

idxs = np.array([1, 3, 2])
vals = np.zeros((idxs.size, idxs.max()+1))
vals[np.arange(idxs.size), idxs] = 1

但我想将其推广到k-hot编码(其中 vals 将相同,但每行可以包含k个)。

不幸的是,我不知道如何为每行的多个col建立索引。我试过了 vals[0:2, [[0, 1], [3]] 从第一行选择第一列和第二列,从第二行选择第三列,但不起作用。

1 回复 | 直到 7 年前

Mazdak 7 年前

从第一行选择第一列和第二列,从第二行选择第三列

您只需在单独的iterables(tuple,list)中传递相应的行和列:

In [9]: a
Out[9]: 
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])

In [10]: a[[0, 0, 1],[0, 1, 3]]
Out[10]: array([0, 1, 8])

推荐文章

serlingpa · 如何准备我的数据以避免无法推断频率

1 年前

Guillaume · 使用操作从Python列表创建numpy数组

2 年前

user19657580 · 在Python中打印两个numpy数组的列表

2 年前

user19657580 · Python中数组中具有相同元素的索引求和

2 年前

mikanim · 改进二维余弦函数的numpy功能

2 年前

Klimt865 · 在Python中将数组列表转换为列表列表

2 年前

theduker · 计算平均绝对误差时,If语句中赋值前引用的局部变量

2 年前

Lynn · 如果列包含Python中的特定字符串,则从列中删除值

2 年前

JasonX · 运行减法计算

2 年前

Jan Hrubec · 选择numpy数组的前n个元素

2 年前