选择数组中大于零的值的索引和

Question

我有两个 np arrays（A，B）。 A中的一些值是0。

我想选择 A 中大于零的值。

并选择 B 中的索引。

最后，将值保存在字典中（我不知道这是否可能）。

这是我想要的 A 和 B 以及 dic。

import numpy as np
A = [[20, 30, 40, 50], [0, 4, 0, 0], [1, 2, 4,0]]
B = [[1, 2, 3, 4],[5, 14, 7, 10], [7, 1, 3,20] ]

dic = ['1': [20, 30, 40, 50], [1, 2, 3, 4],'2': [ 4], [ 14],'3':[1, 2, 4], [7, 1, 3] ] #which I want

有人可以帮我解决这个问题吗？ 太感谢了

Answer 1

是的，可以这样做。 如下所示：

A = [[20, 30, 40, 50], [0, 4, 0, 0], [1, 2, 4, 0]]
B = [[1, 2, 3, 4], [5, 14, 7, 10], [7, 1, 3, 20]]

result_dict = {}
for index, item in enumerate(A):
    result_dict[index] = [[elem for elem in A[index] if elem>0], [elem for elem in B[index] if elem>0]]

print(result_dict)
# OUTPUT
# {0: [[20, 30, 40, 50], [1, 2, 3, 4]], 1: [[4], [5, 14, 7, 10]], 2: [[1, 2, 4], [7, 1, 3, 20]]}

我理解有时很难理解当你是新手时如何进行。 在使用像 np. 例如，您可以使用基本的 python 规则轻松遍历列表并使用索引来执行此操作。

我在这个例子中使用了 enumerate() 和列表推导，你也可以使用其他方法！

快乐编码:)