[英]Identifying whether a list item is in a string
我正在尝试创建一个嵌套循环序列,它查看一系列停用词和字符串列表,并确定每个停用词是否在每个列表项中。 理想情况下,我希望能够将每个字符串中存在的单词添加到新列中,并将它们全部从字符串中删除。
有人有提示吗? 我的循环顺序错误吗?
def remove_stops(text, customStops):
"""
Removes custom stopwords.
Parameters
----------
text : the variable storing strings from which
stopwords should be removed. This can be a string
or a pandas DataFrame.
customStops : the list of stopwords which should be removed.
Returns
-------
Cleansed lists.
"""
for item in text:
print("Text:", item)
for word in customStops:
print("Custom Stops: ", word)
if word in item:
print("Word: ", word)
#Add word to list of words in item
#Remove word from item
这是您可以执行的操作:
def remove_stops(text, customStops):
found = {k:[] for k in text} # Dict for all found stopwords in text
for i,item in enumerate(text):
for word in customStops:
text[i] = text[i].replace(word,'') # Remove all stopwords from each string, if the stopword is not in, the replace will just leave it as it is
if word in item:
found[item].append(word)
return text, found
text = ['Today is my lucky day!',
'Tomorrow is rainy',
'Please help!',
'I want to fly']
customStops = ['help', 'fly']
clean, found = remove_stops(text, customStops)
print(clean)
print(found)
Output:
['Today is my lucky day!',
'Tomorrow is rainy',
'Please !',
'I want to ']
{'Today is my lucky day!': [],
'Tomorrow is rainy': [],
'Please help!': ['help'],
'I want to fly': ['fly']}
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.