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

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

---

---

附录：根据您在此处评论中概述的目的，我会考虑一个仅返回有趣 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'
]

Answer 2

试试这个：

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

    return status, ret

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

Answer 3

您必须将“全局”关键字添加到您的函数中。只需在其最顶部添加一行“全局 url”即可。同时删除参数。

您尝试做的是“副作用”。该函数修改其范围之外的变量。这可能会让读者感到头疼，尤其是当您将函数命名为“url_is_image”时，读者只会期望检查而不是修改。

我建议将副作用移动到一个单独的函数“fix_imgur_jpg_url”，该函数将 url 作为参数并返回固定的 url。

Answer 4

因为当您将url 传递给url_is_image 时，它正在创建一个新的url 对象。修改您的退货声明。不要使用全局变量

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

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

或者您可以将return False 替换为return None，具体取决于接下来会发生什么。

Answer 5

您必须返回您在函数中所做的更改才能生效。

编辑：遵循不返回不同类型的编程实践，之前的代码已被修改。

def url_is_image(url):
    exit_code = 1
    if url.endswith(('.jpg', '.png')):
        pass #exit_code still 1.
    elif 'imgur' in url:
        url += '.jpg' #exit_code still 1.
    else:
        exit_code = 0 #url does not match requirements
    return url, exit_code

u1 = 'http://i.imgur.com/jtboy'
u1, stat = url_is_image(u1)
if stat: #url has been modified or has required extension
    print (u1)