确定两个 python 系列或数据帧中的额外值

Question

给定两个数据框，我将一些唯一值提取到 groupby 中，然后我想比较两个唯一值并找到不同的值。

unit1 = ["U1", "U2", "U1", "U2", "U1", "U2"]
unit2 = ["U1", "U2", "U1", "U2", "U1", "U2, "U3"]
count1 = [2,4,6,8,10,12]

df = pd.DataFrame({'Unit': unit1,
                   'Count': count1})
df2 = pd.DataFrame({'Unit': unit2,
                   'Count': count1})

units_in_1 = df.groupby(['Unit'])
unit1_list = units_in_1['Unit'].unique()

units_in_2 = df2.groupby(['Unit'])
unit2_list = units_in_2['Unit'].unique()

我最终想要的是 U3，这样我就可以返回并在 df2 中找到它的所有实例。

unit1_list 是一个系列，我似乎无法工作。

根据this answer 创建一个列表并进行交集失败，著名的 numpy 不可散列

如果我尝试减法，我会得到类型错误不支持 str 和 str 的操作数

请帮忙。

Answer 1

您可以使用 isin() 和否定 (~) 运算符来实现此目的。

>>> stuff_in_df2_but_not_in_df1 = df2[~df2.Unit.isin(df1.Unit)]['Unit'].unique()
['U3']

这行代码有点晦涩，告诉 Pandas 给你 df2['Unit'] 中没有在 df1['Unit'] 中的所有项目。

如果您希望它在两个方向上都起作用（即，您想要一个在 df1 中但不在 df2 中的东西的列表和在 df2 中但不在df1，你可以使用set.symmetric_difference()。

如果df1['Unit'] 包含U1, U2, U4 并且df2['Unit'] 包含U1, U2, U3，则以下代码将为您提供包含{'U3', 'U4'} 的set()。

>>> set(df1.Unit.unique()).symmetric_difference(set(df2.Unit.unique()))
{'U3', 'U4'}

Answer 2

希望我能理解您的问题。

如果您选择任何不等于 df['Unit] 的 df2['Unit] 值，它会起作用吗？

df2[df.Unit != df2.Unit ]

输出：

Answer 3

IIUC 然后你可以创建一个唯一值的set 然后调用difference：

In [161]:
combined = set(df['Unit'].unique().tolist()+df2['Unit'].unique().tolist())
combined

Out[161]:
{'U1', 'U2', 'U3'}

In [162]:    
combined.difference(df['Unit'].unique())

Out[162]:
{'U3'}