python写一个函数,输入参数为图片的url,要求下载图片
时间: 2024-10-27 09:07:01 浏览: 9
在Python中,你可以使用requests库来获取URL内容,然后使用io库处理文件流,将网络上的图片下载到本地。这里是一个简单的示例函数,它会下载指定URL的图片:
```python
import requests
from io import BytesIO
def download_image(url):
# 发送HTTP请求并获取响应
response = requests.get(url)
# 检查请求是否成功,状态码为200表示成功
if response.status_code == 200:
# 将响应内容转化为BytesIO对象
image_data = BytesIO(response.content)
# 解析文件名(如果有的话)
file_name = url.split("/")[-1] if "?" not in url else url.split("?")[0].split("/")[-1]
# 写入文件
with open(file_name, 'wb') as f:
f.write(image_data.getvalue())
print(f"Image downloaded successfully at {file_name}")
else:
print(f"Failed to download image. Status code: {response.status_code}")
# 使用函数
download_image("http://example.com/image.jpg")
阅读全文