查找二维 numpy 数组中的共同元素

如果我有两个（或更多）二维数组，如何在给定行号的情况下仅获取数组之间的公共元素。例如，我有以下格式的数组：

time, position, velocity, acceleration

我想让两个数组只具有相同的时间元素，所以第 0 行。我可以使用

np.intersect1d(array1[:, 0], array2[:, 0])

它给出了所有公共时间，但我想从 array1/2 中提取所有匹配的行/列或删除非公共时间元素。到底array1 and array2将具有完全相同的尺寸，所以我可以去：

pos_difference = array1[:, 1] - array2[:, 1]

数组的大小可以不同，例如：

array1 = [[1, 100.0, 0.0, 0.0], [2, 110.0, 0.0, 0.0], [3, 120.0, 0.0, 0.0]]
array2 = [[1, 101.0, 0.0, 0.0], [3, 119, 0.0, 0.0]]

我只想提取公共时间元素，因此 array1 和 array2 仅在 Time=1 和 Time=3 时包含，因为这些是公共时间元素。然后我可以去：

pos_difference = array1[:, 1] - array2[:, 1]

这将是两个数组同时之间的位置差异：

# First row will be when time=1 and second row will be when time=3
pos_difference = [[0, -1, 0.0, 0.0], [0, 1, 0.0, 0.0]]

如果你有这些数组：

import numpy as np
array1 = np.array([[1, 100.0, 0.0, 0.0], [2, 110.0, 0.0, 0.0], [3, 120.0, 0.0, 0.0]])
array2 = np.array([[1, 101.0, 0.0, 0.0], [3, 119, 0.0, 0.0]])

正如你所说，你可以使用np.intersect1d要获得交集，唯一剩下的就是对数组进行索引：

intersect = np.intersect1d(array1[:, 0], array2[:, 0])

array1_matches = array1[np.any(array1[:, 0] == intersect[:, None], axis=0)]
array2_matches = array2[np.any(array2[:, 0] == intersect[:, None], axis=0)]

然后你可以减去它们：

>>> array1_matches - array2_matches
array([[ 0., -1.,  0.,  0.],
       [ 0.,  1.,  0.,  0.]])

这假设您的时间是唯一且排序的。如果它们未排序，您可以先对它们进行排序：

>>> array1 = array1[np.argsort(array1[:, 0])]
>>> array2 = array2[np.argsort(array2[:, 0])]

如果时间不独特，我不知道你想如何处理这个问题，所以我无法给你建议。