Python:字符串操作在函数内部不起作用

Question

代码如下:

def url_is_image(url):
    if url[-4:] in ['.jpg', '.png']:
        return True
    elif 'imgur' in url:
        url += '.jpg'
        return True
    else:
        return False

u1 = 'http://i.imgur.com/jtboy'
url_is_image(u1)
print u1

输出是'http://i.imgur.com/jtboy ' 没有 .jpg 扩展名。关于为什么我的函数(第 5 行)无法将 .jpg 添加到字符串的任何想法？谢谢

Answer 1

参数 url 的范围是函数调用的局部范围。对其赋值不会影响函数外的任何变量或值。

变量u1被赋值一次。函数调用不影响它。然后打印出来。

因为你没有测试函数返回值的真实性，你可以这样重写它:

def url_is_image(url):
    if url[-4:] in ['.jpg', '.png']:
        return url
    elif 'imgur' in url:
        return url + '.jpg'
    else:
        return url

u1 = 'http://i.imgur.com/jtboy'
u1 = url_is_image(u1)
print u1

但是，函数名称在这种情况下具有误导性。由于我不知道您希望如何真正使用该功能，因此很难提出更好的解决方案。也许您想返回一个包含 bool 值和修改后的字符串的二元组。

---

---

附录:根据您在此处评论中概述的目的，我会考虑一个仅返回有趣 URL 的函数。例如:

def consider_url(url):
    if url[-4:] in ['.jpg', '.png']:
        return url
    elif 'imgur' in url:
        return url + '.jpg'
    else:
        return None

然后您可以使用它来转换有趣的 URL 并忽略不感兴趣的 URL。

for url in some_list:
    url = consider_url(url)
    if url is None:
        continue
    print(url)

另一种方法是编写一个只为您提供感兴趣的 url 的生成器:

def interesting_urls(iter):
    for url in iter:
        if url[-4:] in ['.jpg', '.png']:
            yield url
        elif 'imgur' in url:
            yield url + '.jpg'

然后您可以更简洁地处理您的网址:

for url in interesting_urls(some_list):
    print(url)

这些示例假设您已经以某种方式检索了列表或其他可迭代对象中的 URL。您可以使用如下所示的简单列表进行尝试。尽管列表中的大部分字符串都不是有效的 URL，但它们确实执行了上述函数的逻辑。

some_list = [
    "foo", "bar.jpg", "something/imgur/baz", "uninteresting", 'http://i.imgur.com/jtboy'
]