Проблема при загрузке URL с фрейма данных - PullRequest
0 голосов
/ 22 апреля 2020

Я хотел бы загрузить все URL-адреса в моем CSV-файле с именем 'rlth.csv'. Файл CSV имеет заголовок внутри. Вот заголовок файла CSV:

    identifier  link    likes_count company
0   2292512316069378197 https://www.instagram.com/p/B_Qo84ihfiV 9608    Ralph Lauren
1   2292462538514040606 https://www.instagram.com/p/B_QdohlBQce 9462    Ralph Lauren

import csv

with open('rlth.csv', 'r', newline='') as file:
    has_header = csv.Sniffer().has_header(file.read(1024))
    file.seek(0)  # Rewind.
    reader = csv.reader(file)
    if has_header:
        next(reader)  # Skip header row.
    csvrows = csv.reader(file, delimiter=',', quotechar='"')
    for row in csvrows:
        filename = row[0]
        url = row[2]
        print(url)
        result = requests.get(url, stream=True)
        if result.status_code == 200:
            image = result.raw.read()
            open("{}.jpg".format(filename),"wb").write(image)

Решено.

...