擅长:python、mysql、java
<p>使用<a href="https://pandas.pydata.org/docs/reference/api/pandas.Series.str.startswith.html" rel="nofollow noreferrer">startswith</a>和<a href="https://predictivehacks.com/pandas-how-to-assign-values-based-on-multiple-conditions-of-different-columns/" rel="nofollow noreferrer">np.select</a></p>
<pre><code>conditions = [df['A'].str.startswith('INKA') | df['B'].str.startswith('INKA') | df['A'].str.startswith('IDKA') | df['B'].str.startswith('IDKA'),
df['A'].str.startswith('INAP') | df['B'].str.startswith('INAP') | df['A'].str.startswith('IDAP') | df['B'].str.startswith('IDAP'),
df['A'].str.startswith('INRJ') | df['B'].str.startswith('INRJ') | df['A'].str.startswith('IDRJ') | df['B'].str.startswith('IDRJ')]
choices = ['KAR','AP', 'RAJ']
df['C'] = np.select(conditions, choices, default=None)
</code></pre>
<p>结果df:</p>
<p><a href="https://i.stack.imgur.com/G5CHW.png" rel="nofollow noreferrer"><img src="https://i.stack.imgur.com/G5CHW.png" alt="enter image description here"/></a></p>