Python 如何找到一个值与熊猫中的多个值相关联
我在熊猫中有以下数据帧Python 如何找到一个值与熊猫中的多个值相关联,python,pandas,Python,Pandas,我在熊猫中有以下数据帧 code tank product 1234 1 MS 1234 2 HS 1234 1 HS 1234 1 HS 1235 1 MS 1235 1 HS 1235 1 MS 1245 1 MS 1245 2
code tank product
1234 1 MS
1234 2 HS
1234 1 HS
1234 1 HS
1235 1 MS
1235 1 HS
1235 1 MS
1245 1 MS
1245 2 HS
我想在上面的数据框中找出有多少储罐有多个产品与之关联,例如代码1234储罐1也有Ms和HS
上述数据框中有2个案例
我想要的数据帧是
code tank flag
1234 1 yes
1234 2 no
1235 1 yes
1245 1 no
1245 2 no
我怎样才能在熊猫身上做呢 用于计算每个组的唯一值:
df = df.groupby(['code','tank'])['product'].nunique().reset_index()
print (df)
code tank product
0 1234 1 2
1 1234 2 1
2 1235 1 2
3 1245 1 1
4 1245 2 1
然后使用提取列并通过设置值
df['flag'] = np.where(df.pop('product') == 1, 'no', 'yes')
print (df)
code tank flag
0 1234 1 yes
1 1234 2 no
2 1235 1 yes
3 1245 1 no
4 1245 2 no