在不同大小的Python的2D数组中查找通用元素

时间:2019-08-22 20:38:53

标签: python arrays numpy

我有2个numpy数组:一个这样的数组(称为a):

[ Frames  value
  1        0.3, 
  2        0.21, 
  3        0.51, 
  4         0.27 
....] 

和另一个一维数组(称为b),其中包含来自第二列的数据。我想找到a中存在的b值,然后在b中添加相对的Frames值。 我正在使用numpy

我已经从.dat创建了np数组,但是从那里我不知道如何进行

这是一个数组

[ Frames  value
  1       0.3, 
  2        0.21,
  3        0.51, 
  4         0.27 
....]

1 个答案:

答案 0 :(得分:1)

如果可以为每个数组的元素创建一个集合,则可以将pythons Set类与instersection()方法一起使用,以在两个集合中查找值。否则,您可以将每个数组转换为pandas数据帧,并在该列上使用熊猫的merge来查找公用元素。