擅长:python、mysql、java
<pre><code>def theme(x):
category = []
for i in x:
if 'AC' in i:
category.append('AC problem')
elif 'insects' in i:
category.append('Cleanliness')
elif 'clean' in i:
category.append('Cleanliness')
elif 'food' in i:
category.append('Food Problem')
elif 'delay' in i:
category.append('Train Delayed')
else:
category.append('None')
return category
categories = theme(['bla bla AC bla','bla insects bla'])
print(categories)
</code></pre>