减法等值itertools.产品()

Question

我是一名大学生，正在做一个项目，分析一些大型数据集。你知道吗

简化我的问题，我有两组点，在矩阵“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），那么我就可以很简单地从那里开始。你知道吗

有人知道有什么东西可以提供这种功能吗？ 或者有没有其他方法不需要缓慢的迭代就可以得到所有这些点之间的夹角？你知道吗

提前谢谢

亚历克斯

Answer 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]))