python - 如何比较两个不同长度的 Python Pandas 系列？

Question

我有两个不同长度的系列，我想获得两个系列中两个索引和数量相同的索引。

这是系列:

ipdb> s1
s1
000007720          2000.00
group1            -3732.05
group t3           2432.12
group2           -38147.87
FSHLAJ           -36711.09
EWkayuwo             -3.22
Name: amount, dtype: float64
ipdb> s2
s2
000007720                 2000.00
group1                   -3732.05
group z                  12390.00
group y                  68633.43
group x                     25.00
group w                   3913.00
group v                 -12750.50
group u                    -53.49
group t                  -7500.00
group s                  -1575.82
group r                    -10.00
group q                   1800.00
group p                  -4510.34
EWFhjkaQU                  455.96
group2                  -38147.87
FSHLAJ                  -36711.09
GEKWJ                        5.54
Name: amount, dtype: float64

当我尝试比较它们时，我得到:

ipdb>s1 == s2
*** ValueError: Series lengths must match to compare

我怎样才能实现我的目标？

Answer 1

您想使用 isin :

In [121]:

s2[s2.isin(s1)]
Out[121]:
000007720
group1    -3732.05
group2   -38147.87
FSHLAJ   -36711.09
Name: 2000.00, dtype: float64

我不知道你想用哪种方式进行比较，这里是另一种方式:

In [122]:

s1[s1.isin(s2)]
Out[122]:
000007720
group1    -3732.05
group2   -38147.87
FSHLAJ   -36711.09
Name: 2000.00, dtype: float64

尝试执行 s1 == s2 的问题在于比较不同长度的系列或数组没有意义。

如果您希望索引也匹配，则将其添加为条件:

In [131]:

s1[(s1.index.isin(s2.index)) & (s1.isin(s2))]
Out[131]:
000007720
group1    -3732.05
group2   -38147.87
FSHLAJ   -36711.09
Name: 2000.00, dtype: float64