如何获取数组中非零项的索引并使用此索引从另一个数组或列表中获取另一个值

Question

我有一个项目，您必须在数组中选择一个非零项并获取其索引。使用这个索引，你必须在另一个数组或列表中找到它对应的值。

例如：

best_chr = [[1 1 0]] # numpy array
activity = [2, 3, 4]

代码如下所示：

chosen_act = [activity[item] for item in range(len(best_chr)) if item != 0]
print(chosen_act)

我打算做的是在best_chr中找到non-zero项目/值的index [1, 1 for the example above]。然后，使用这个索引，在activity 中找到它的对应值。

Expected output:
chosen_act = [2, 3]

问题是，使用上面的代码，我收到以下错误：

TypeError: object of type 'numpy.int32' has no len()

任何帮助将不胜感激！谢谢！

Answer 1

你可以使用zip:

# best_chr.flatten since it looks like 2-d array
[i for i, j in zip(activity, best_chr.flatten()) if j]

如果需要使用索引，使用numpy.flatnonzero:

[activity[i] for i in np.flatnonzero(best_chr)]

输出：

[2, 3]