Prompt Example
너는 Python 데이터 생성기 역할을 해주었으면 해. 고양이 사진이 포한된 Csv파일을 생성할 수 있어? 파일은 Cat파일에 저장해줘.
Result
네, 다음과 같이 고양이 사진이 포함된 CSV 파일을 생성하는 코드를 작성할 수 있습니다. 이 코드는 현재 작업 디렉토리에서 `cat_photos.csv` 파일을 생성하며, 각 사진에 대한 링크와 해당하는 id 값을 포함합니다.
import os
import csv
#고양이 사진이 저장된 폴더 경로
folder_path = './Cat/'
#폴더 내의 고양이 사진 파일 리스트
cat_images = [os.path.join(folder_path, f) for f in os.listdir(folder_path) if f.endswith('.jpg')]
#CSV 파일 생성
with open('cat_photos.csv', mode='w', newline='') as file:
writer = csv.writer(file)
writer.writerow(['id', 'image_path'])
for i, image_path in enumerate(cat_images):
writer.writerow([i+1, image_path])
위 코드에서는 `os` 모듈을 사용하여 고양이 사진이 저장된 폴더 내의 `.jpg` 확장자를 가진 파일 리스트를 가져와 `cat_images` 리스트에 저장합니다. 그리고 이 리스트를 이용하여 각 사진에 대한 파일 경로와 해당하는 id 값을 포함하는 `cat_photos.csv` 파일을 생성합니다. 이 코드를 실행하려면 `./Cat/` 폴더가 존재하고 폴더 내에 고양이 사진이 `.jpg` 확장자를 가지고 있어야 합니다.