Python – 根据url下载图片的几种方式整理

在写爬虫的时候经常需要根据图片的url直链下载图片到本地，本文整理了根据图片url下载图片的一些方式，大家可以参考参考。

1 python根据url下载图片

1.1 使用urllib下载图片

我们可以使用urllib的urlretrieve函数保存图片，示例代码如下

from urllib.request import urlretrieve

def urllib_download_image(image_url, image_path):
    urlretrieve(image_url, image_path)

if __name__ == '__main__':
    image_url = 'https://img-blog.csdnimg.cn/20200924162143340.jpg'
    image_path = './image.jpg'
    urllib_download_image(image_url, image_path)

1.2 使用request库下载图片

使用request库有两种方式，一种是直接下载，另一种分块保存，一般来说如果图片文件大小在5mb以内就可以直接下载，如果超过5mb则建议使用分块下载的方式，避免内存溢出。

request直接下载

import requests

def request_direct_download_image(image_url, image_path):
    res = requests.get(image_url)
    with open(image_path, 'wb') as f:
        f.write(res.content)

if __name__ == '__main__':
    image_url = 'https://img-blog.csdnimg.cn/20200924162143340.jpg'
    image_path = './image.jpg'
    request_direct_download_image(image_url, image_path)

request分块下载

import requests

def request_chunk_download_image(image_url, image_path):
    res = requests.get(image_url)
    with open(image_path, 'wb') as f:
        for chunk in res.iter_content(chunk_size=1024):
            f.write(chunk)

if __name__ == '__main__':
    image_url = 'https://img-blog.csdnimg.cn/20200924162143340.jpg'
    image_path = './image.jpg'
    request_chunk_download_image(image_url, image_path)

这里补充一句，如果我们需要下载的是视频文件，一般来说视频文件都比较大，所以也建议使用分块下载的方式下载视频，代码可参考上面分块下载图片的代码。