基于字典存在性的numpy数组过滤

这是一种方法 numpy.in1d :

keys = list(k.keys())
res = x[np.in1d(x[:, -1], keys)]

print(res)

[[1 2 1]
 [3 2 3]]

我会做一些事情,比如:

import numpy as np
x = np.array([[1, 2, 1], [4, 5, 7], [3, 2, 3]])
k = {1: None, 2: None, 3: None}
classes = [i for i in x[:,-1] if i in k.keys()]
classes = np.array(classes)
print(classes)

这应该返回1和3,而不是7,因为我们只查找x中的最后一行数据值。这将返回k中的值列表,如果您选择,可以将其形成数组。