【发布时间】:2018-07-10 11:16:04
【问题描述】:
给定一个 numpy 数组,它可以是满足给定条件的数组元素索引的子集。如何从生成的索引对创建三元组(或四元组、五元组……)的元组?
在下面的示例中,pairs_tuples 等于 [(1, 0), (3, 0), (3, 1), (3, 2)]。 triplets_tuples 应该是 [(0, 1, 3)] 因为它的所有元素(即 (1, 0), (3, 0), (3, 1))都有满足条件的成对值,而 (3, 2)才不是。
a = np.array([[0. , 0. , 0. , 0. , 0. ],
[0.96078379, 0. , 0. , 0. , 0. ],
[0.05498203, 0.0552454 , 0. , 0. , 0. ],
[0.46005028, 0.45468466, 0.11167813, 0. , 0. ],
[0.1030161 , 0.10350956, 0.00109096, 0.00928037, 0. ]])
pairs = np.where((a >= .11) & (a <= .99))
pairs_tuples = list(zip(pairs[0].tolist(), pairs[1].tolist()))
# [(1, 0), (3, 0), (3, 1), (3, 2)]
如何到达下方?
triplets_tuples = [(0, 1, 3)]
quadruplets_tuples = []
quintuplets_tuples = []
【问题讨论】:
-
你能更准确地定义“它的所有元素”吗?你如何从
(0, 1, 3)到[(1, 0), (3, 0), (3, 1)]?为什么(0, 1)没有出现在那个集合中? -
你的输入矩阵总是下三角吗?
-
感谢您的 cmets 和问题。输入矩阵可以被认为是相关矩阵的下三角。鞋面只是对称的,因此对 (1, 0) 等价于/可与对 (0, 1)) 交换。我想找到所有满足成对条件的索引的三元组(或 n-let,其中 n>2),即在示例中,索引 [(0, 1, 3)] 是唯一有效的三元组,因为所有对((1, 0), (3, 0), (3, 1)) 三元组内满足条件。
-
np.transpose(pairs)简化了获取对。
标签: python arrays numpy tuples pairwise