itertools.product() 的减法等价物答案

【问题标题】：Subtraction equivalent of itertools.product()itertools.product() 的减法等价物
【发布时间】：2018-11-17 14:00:06
【问题描述】：

我是一名大学生，正在从事一个分析一些大型数据集的项目。

简化我的问题，我有两组点，在矩阵“A”和“B”中这样：

A = [[x1, y1], [x2, y2],...] and B = [[x'1, y'1], [x'2, y'2],...]

我想创建一个函数，它输出一个带有元素的矩阵 C：

Cij  = atan((y'i - yj)/(x'i - xj))

基本上，连接任意两点的线所对的角度（wrt x.axis），每个列表中的一个。数据集足够大，因此嵌套的 FOR 循环不是一种选择。

目前的尝试将我引向了 itertools 产品功能。如果有一个等价物在元素之间提供减法（即 y'i-yj ），那么我可以很简单地从那里开始。

有人知道可以提供此功能的东西吗？或者也许有任何其他方法可以在没有缓慢迭代过程的情况下实现所有这些点之间的角度？

提前致谢，

亚历克斯

【问题讨论】：

【解决方案1】：

使用 numpy 进行这些计算

import numpy as np

A = np.array(A)
B = np.array(B)

C = np.arctan((B[:, None,  1] - A[:, 1]) / (B[:, None, 0] - A[:, 0]))

【讨论】：