how to write a csv file code example
Example 1: python save as csv
import numpy as np
np.savetxt('data.csv', (col1_array, col2_array, col3_array), delimiter=',')
Example 2: Write a table to CSV file python
header = [re.sub(' +',' ',i[0][:-1].replace('\n', ' ')) for i in optionsTable[0]]
with open('test.csv', 'w') as fp:
writer = csv.writer(fp, delimiter=',')
writer.writerow(header)
for row in optionsTable:
writer.writerow([i[1] for i in row])
Example 3: 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)