在单个单词字符串PYTHON中查找数字字符

Question

在CSV的文本字段中有多种值

有些值是这样的 AGM00鲍德温 AGM00BOUCK公司

但是，有些是重复的，将名称改为 AGM00BOUCK01型 AGM00COBDEN01型 AGM00COBDEN02型

我的目标是为不包含数字后缀的值编写一个特定的ID

这是目前为止的代码

prov_count = 3000
prov_ID = 0
items = (name, x, y)
xy_tup = tuple(items)

if "*1" not in name and "*2" not in name:
    prov_ID = prov_count + 1
else:
prov_ID = ""

看起来通配符在这里不是合适的方法，但我似乎找不到合适的解决方案。你知道吗

Answer 1

可以使用切片查找元素的最后2个字符，然后检查它是否以'01'或'02'结尾：

l = ["AGM00BALDWIN", "AGM00BOUCK", "AGM00BOUCK01", "AGM00COBDEN01", "AGM00COBDEN02"]

for i in l:
    if i[-2:] in ('01', '02'):
        print('{} is a duplicate'.format(i))

输出：

AGM00BOUCK01 is a duplicate
AGM00COBDEN01 is a duplicate
AGM00COBDEN02 is a duplicate

或者另一种方法是使用str.endswith方法：

l = ["AGM00BALDWIN", "AGM00BOUCK", "AGM00BOUCK01", "AGM00COBDEN01", "AGM00COBDEN02"]

for i in l:
    if i.endswith('01') or i.endswith('02'):
        print('{} is a duplicate'.format(i))

所以您的代码如下所示：

prov_count = 3000
prov_ID = 0
items = (name, x, y)
xy_tup = tuple(items)

if name[-2] in ('01', '02'):
    prov_ID = prov_count + 1
else:
    prov_ID = ""