代码之家 › 专栏 › 技术社区 › Aurélien Pierre

Numpy快速布尔交互矩阵

numpy python

-1

Aurélien Pierre · 技术社区 · 2 年前

我有两个整数向量,一个很长 document (1E5到1E7元素),还有一个相当短 query (通常为5-8个元素)。我想创建一个2D布尔矩阵 (i, j) 这使得 1 哪里 document[i] == query[j] 否则 0 。例如:

文档\查询	5.	2.	4.
2.	0	1.	0
8.	0	0	0
3.	0	0	0
4.	0	0	1.
...

有没有一种快速的方法来使用Numpy,即不使用Python循环?(此处不能使用熊猫)

1 回复 | 直到 2 年前

Axel Donath 2 年前

你可以利用numpy广播:

import numpy as np

document = np.random.randint(0, 5, (10,))
query = np.array([2, 3, 5])

(document[:, np.newaxis] == query)

打印内容:

array([[False, False, False],
       [False, False, False],
       [False,  True, False],
       [False, False, False],
       [False, False, False],
       [False,  True, False],
       [ True, False, False],
       [False, False, False],
       [False, False, False],
       [ True, False, False]])

我希望这能回答你的问题!

推荐文章

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

1 年前

unfolx · numpy数组不等式的执行时间

1 年前

mchaudh4 · 用numpy表示三对角矩阵

1 年前

Geremia · 2D NumPy数组+1D数组?

1 年前

LMC · Numpy数组布尔索引以获取包含元素

1 年前

HJA24 · 根据条件用值正向填充Numpy矩阵/掩码

1 年前

alpelito7 · 如何只获取numpy数组中每个递增值的第一次出现?

1 年前

Amarth Gûl · 找到一组向量的最近收敛点

1 年前

Mr. W · numpy.divide是函数、类还是其他什么?

1 年前

Mr. W · 为什么numpy.array在编辑内部数据时如此缓慢?

1 年前