python - 通过字符串列中的最后 3 个字符选择行

Question

我有这个数据框

      name               year ...
0     Carlos - xyz       2019
1     Marcos - yws       2031
3     Fran - xxz         2431
4     Matt - yre         1985
...

我想创建一个名为 type 的新列。如果此人的名字以“xyz”或“xxz”结尾，我希望类型为“big”

所以，它应该是这样的:

      name               year   type
0     Carlos - xyz       2019    big
1     Marcos - yws       2031  
3     Fran - xxz         2431    big
4     Matt - yre         1985
...

有什么建议吗？

Answer 1

选项 1
使用 str.contains 生成掩码:

m = df.name.str.contains(r'x[yx]z$')

或者，

sub_str = ['xyz', 'xxz']
m = df.name.str.contains(r'{}$'.format('|'.join(sub_str)))

现在，您可以使用 np.where 创建您的列，

df['type'] = np.where(m, 'big', '')

或者，用loc代替np.where；

df['type'] = ''
df.loc[m, 'type'] = 'big'

df
           name  year type
0  Carlos - xyz  2019  big
1  Marcos - yws  2031     
3    Fran - xxz  2431  big
4    Matt - yre  1985     

---

选项 2
作为替代方案，考虑 str.endswith + np.logical_or.reduce

sub_str = ['xyz', 'xxz']
m = np.logical_or.reduce([df.name.str.endswith(s) for s in sub_str])

df['type'] = ''
df.loc[m, 'type'] = 'big'

df
           name  year type
0  Carlos - xyz  2019  big
1  Marcos - yws  2031     
3    Fran - xxz  2431  big
4    Matt - yre  1985