python - NumPy ，1 :M joins on Arrays

Question

我想知道是否有办法加入 numpy 数组。

示例:

array1 = [[1,c,d], [2,a,b], [3, e,f]]
array2 = [[2,g,g,t], [1,alpha, beta, gamma], [1,t,y,u], [3,dog, cat, fish]]

我需要加入这些数组，但 Numpy 文档说如果记录不唯一，函数将失败或返回未知结果。

有人有任何示例可以在 numpy 数组上进行 1:M 连接而不是 1:1 连接吗？另外，我知道我的示例采用正确的 numpy 格式，但这只是为了提供一个总体思路。

Answer 1

您愿意实现的目标看起来更像是基于两个输入数组的新嵌套列表。

将它们视为列表:

list1 = [[1,'c','d'], [2,'a','b'], [3, 'e','f']]
list2 = [[2,'g','g','t'], [1,'alpha', 'beta', 'gamma'], [1,'t','y','u'], [3,'dog', 'cat', 'fish']]

您可以通过以下方式构建您想要的结果:

result = [i+j[1:] for i in list1 for j in list2 if i[0]==j[0]]

看起来像这样:

[[1, 'c', 'd', 'alpha', 'beta', 'gamma'],
 [1, 'c', 'd', 't', 'y', 'u'],
 [2, 'a', 'b', 'g', 'g', 't'],
 [3, 'e', 'f', 'dog', 'cat', 'fish']]