用python批量下载csv中的链接为图片

用python批量下载csv中的链接为图片

问的文心一言：

我得到了一个csv文件，内容是图片链接，也就是说点击这个链接可以得到一张图片，我该怎么将这个csv文件中的链接批量下载为图片？

效果很nice

import csv  
import requests  
from io import BytesIO  
import os  

# 定义下载图片的文件夹路径  
output_folder = 'D:/25678/Desktop/images'  
  
# 检查并创建下载文件夹（如果不存在）  
if not os.path.exists(output_folder):  
    os.makedirs(output_folder)  
i = 0
# 读取CSV文件  
with open('D:/25678/Desktop/photo.csv', mode='r') as file:  
    reader = csv.reader(file)  
    for row in reader:  
        # 从CSV文件中读取每行的链接  
        link = row[0]  

        #print(link)
  
        # 发送GET请求获取图片  
        response = requests.get(link)  
  
        # 检查请求是否成功  
        if response.status_code == 200:  
            # 将图片保存到磁盘  
            image_name = 'image_{}.jpg'.format(i)  # 假设图片文件名与链接中的文件名相同  
            with open(os.path.join(output_folder, image_name), 'wb') as image_file:  
                image_file.write(response.content)  
        else:  
            print(f"Failed to retrieve image from {link}. Status code: {response.status_code}")
        i = i + 1