python中的numpy索引[重复]

Question

这是交易，

idx_arr = [0,3,5,7];
tgt_arr = [
[0,3,3,5,5,6,6],
[1,1,3,1,1,3,3],
[2,4,6,8,1,2,9]]

我想创建一个看起来像 bool 类型的新数组。我也尝试使用集合，但 numpy.ndarrays 是不可散列的类型。新矩阵看起来像

final_arr = [
[t,t,t,t,t,f,f],
[f,f,t,f,f,t,t],
[f,f,f,f,f,f,f]]

提前致谢。

Answer 1

使用基础 Python：

[[True if val in idx_arr else False for val in row] for row in tgt_arr]

结果：

[[True, True, True, True, True, False, False],
[False, False, True, False, False, True, True],
[False, False, False, False, False, False, False]]