how to create a csv file code example

Example 1: create csv file python

# This action requires the 'csv' module
import csv

# The basic usage is to first define the rows of the csv file:
row_list = [["SN", "Name", "Contribution"],
             [1, "Linus Torvalds", "Linux Kernel"],
             [2, "Tim Berners-Lee", "World Wide Web"],
             [3, "Guido van Rossum", "Python Programming"]]

# And then use the following to create the csv file:
with open('protagonist.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(row_list)
# This will create a csv file in the current directory

Example 2: Creating csv file

import pandas as pdimport osimport torchdevice = ("cuda" if torch.cuda.is_available() else "cpu")train_df = pd.DataFrame(columns=["img_name","label"])train_df["img_name"] = os.listdir("train/")for idx, i in enumerate(os.listdir("train/")):    if "cat" in i:        train_df["label"][idx] = 0    if "dog" in i:        train_df["label"][idx] = 1train_df.to_csv (r'train_csv.csv', index = False, header=True)

Tags:

Python Example

Recent Posts