smer_prods是这样的字典:
smer_prods = {
#'vermicelli' : ['vermicelli', 'ragi vermicelli', 'rice vermicelli'],
'ragi vermicelli' : ['ragi vermicelli'],
'rice vermicelli' : ['rice vermicelli'],
'vermicelli jupiter' : ['vermicelli jupiter'],
'lemon & tamarind vermicelli' : ['lemon & tamarind vermicelli'],
'finosta vermicelli' : ['finosta vermicelli-5kg'],
'rosted vermicelli' : ['roasted vermicelli'],
'semiya/vermicelli' : ['semiya / vermicelli 900grams'],
'vermicelli upma' : ['vermicelli upma'],
'vermicelli payasam mix' : ['vermicelli payasam mix'],
'mung bean vermicelli' : ['mung bean vermicelli'],
'red chili' : ['red chilli (lal mirch)','guntur red chilli','red chilly whole(lal mirch)', 'red chilly wg', 'red chilli whole (hot) 1 kg', 'red chilli whole (rich colour) 1 kg'],
'red chili powder' : ['red chilli fresh-kg','red chilli powder (rich colour) 1 kg','red chilli powder (hot) 1 kg','red chilli powder','lal mirch powder','lal mirch powder 100gms', 'lal mirch powder 1kg', 'lal mirch powder 200gms', 'lal mirch powder 500gms'],
'red chilli sauce' : ['red chilli sauce', 'red chilli sauce 200gm pet bottle 48X200gm', 'hot chili sauce'],
'sriraja hot chilli sauce' : ['sriraja hot chilli sauce', 'sriracha hot chilli sauce'],
'mineral water' : ['himalayan orchard pure peach flavoured natural mineral water - 500 ml','himalayan orchard pure strawberry flavoured natural mineral water - 500 ml','himalayan orchard pure apple flavoured natural mineral water - 500 ml','himalayan - the natural mineral water - 500ml bottle', 'himalayan - the natural mineral water - 200ml bottle', 'himalayan - the natural mineral water - 1ltr bottle'],
}
现在,当我执行以下简单代码时:
for x in smer_prods:
mask = df['ITEM NAME'] == x
df1 = df[mask]
print(df1['ITEM NAME'])
它应该仅打印ITEM_NAME列,但它会打印很多不必要的额外信息:
4 rice vermicelli
Name: ITEM NAME, dtype: object
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
5 ragi vermicelli
Name: ITEM NAME, dtype: object
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
1 sriraja hot chilli sauce
Name: ITEM NAME, dtype: object
Series([], Name: ITEM NAME, dtype: object)
Series([], Name: ITEM NAME, dtype: object)
我只想要列表中匹配的项目。我为此写了一个代码:
for x in smer_prods:
mask = df['ITEM NAME'] == x
df1 = df[mask]
item_name = df1['ITEM NAME'].tolist()
print(item_name)
它产生了以下输出: ['米粉']
[]
[]
['ragi vermicelli']
[]
[]
[]
[]
[]
[]
[]
[]
['sriraja hot chilli sauce']
[]
[]
所需的输出:
['rice vermicelli', 'ragi vermicelli', 'sriraja hot chili sauce']
编辑: 如果我这样写:
match = []
for x in smer_prods:
mask = df['ITEM NAME'] == x
if mask.any() == True:
df1 = df[mask]
item_name = df1['ITEM NAME']#note
match.append(item_name)
print(match)
print('-'*80)
它输出:
[4 rice vermicelli
Name: ITEM NAME, dtype: object, 5 ragi vermicelli
Name: ITEM NAME, dtype: object, 1 sriraja hot chilli sauce
Name: ITEM NAME, dtype: object]
答案 0 :(得分:1)
我认为您需要列表中的测试值,因此将列表理解与isin
一起用于字典的测试值:
df = pd.DataFrame({'ITEM NAME':['ragi vermicelli','red chilli (lal mirch)']})
print (df)
ITEM NAME
0 ragi vermicelli
1 red chilli (lal mirch)
#if need matched values
L = [y for k, v in smer_prods.items()
for y in df.loc[df['ITEM NAME'].isin(v), 'ITEM NAME'].tolist()]
print (L)
['ragi vermicelli', 'red chilli (lal mirch)']
#if need matched key of dictionary
L1 = [k for k, v in smer_prods.items()
for y in df.loc[df['ITEM NAME'].isin(v), 'ITEM NAME'].tolist()]
print (L1)
['ragi vermicelli', 'red chili']
循环解决方案:
L = []
for k, v in smer_prods.items():
for y in df.loc[df['ITEM NAME'].isin(v), 'ITEM NAME'].tolist():
print (y)
L.extend([y])
#if need matched key of dictionary
#L.extend([k])
print (L)
['ragi vermicelli', 'red chilli (lal mirch)']