python - 从另一列中获取具有至少一个指定值的值
问题描述
我有这个数据集
custID documentno quantity producttype
1 1281261658 3 accesories
1 1281261658 3 major
1 1281261658 3 accesories
2 0105162445 2 major
2 0105162445 2 major
3 0100848728 2 major
3 0100848728 2 accesories
4 0106075074 1 major
因此,我想以一种我希望 documentno 至少包含一个“附件”的方式过滤数据集。所以我想删除其中只有“主要”的文档。
custID = {1,1,1,2,2,3,3,4}
documentno = {1281261658,1281261658,1281261658,0105162445,0105162445,0100848728,0100848728,0106075074}
quantity = {3,3,3,2,2,2,2,1}
producttype = {'accesories','major','accesories','major','major','major','accesories','major'}
最终结果是:
id documentno quantity producttype
1 1281261658 3 accesories
1 1281261658 3 major
1 1281261658 3 accesories
3 0100848728 2 major
3 0100848728 2 accesories
我尝试了if else,但我做不到。
提前致谢
解决方案
给定pd.DataFrame
df
:
df = pd.DataFrame({'id' : [1,1,1,2,2,3,3,4],
'documentno' : ['1281261658','1281261658','1281261658','0105162445','0105162445','0100848728','0100848728','0106075074'],
'quantity' : [3,3,3,2,2,2,2,1],
'producttype' : ['accesories','major','accesories','major','major','major','accesories','major']})
尝试以下操作:
df.loc[df.documentno.isin(df.documentno[df.producttype.eq('accesories')].unique()),:]
结果是:
id documentno quantity producttype
0 1 1281261658 3 accesories
1 1 1281261658 3 major
2 1 1281261658 3 accesories
5 3 0100848728 2 major
6 3 0100848728 2 accesories
作为过滤器df.documentno[df.producttype.eq('accesories')]
,它返回documentno
具有一个或多个accesories
. 我假设quantity
在数据集中df
总是大于 0。
但是,如果quantity
可能为零:
df = pd.DataFrame({'id' : [1,1,1,2,2,3,3,4],
'documentno' : ['1281261658','1281261658','1281261658','0105162445','0105162445','0100848728','0100848728','0106075074'],
'quantity' : [3,3,3,2,2,2,0,1],
'producttype' : ['accesories','major','accesories','major','major','major','accesories','major']})
只需添加条件df.quantity.gt(0)
:
df.loc[ df.documentno.isin(df.documentno[(df.producttype.eq('accesories') & df.quantity.gt(0))].unique()) ,:]
推荐阅读
- javascript - 如何清除被嘲笑的剧作家回应?
- ios - 将视频上传到云存储时出错,但不是图像 (Swift/Xcode/iOS)
- php - 运算符“===”在对象中使用时无法正确评估,因为它应该 php
- python - 如何从 Treeview / Tkinter (SQLite) 中的不同表中获取不同的值?
- python - 为什么 hist2d 绘制 UNIFORM 密度图?
- python - 我如何压缩所有这些“if”语句?
- azure-devops - 在 Azure DevOps 测试结果中查看单元测试说明
- android - ExposedDropdownMenu 中的字符串以外的其他对象
- javascript - 使用highchart在柱形图中获取一系列的一系列
- python-3.x - 如何在 Python 中将毫秒转换为具有小时、分钟和秒的格式?(超过 24 小时)