擅长:python、mysql、java
<p>使用<code>str.extract</code>从基于字符串的列中获取子字符串</p>
<pre><code>d = {'apple': 'A001', 'ball': 'B099', 'fan': 'F009'}
df['category'] = (
df.descriptions
.str.lower()
.str.extract('(' + '|'.join(d.keys()) + ')')
.squeeze().map(d)
)
</code></pre>