python import csv into array code example
Example 1: python read csv into array
import csv
items = []
with open('file.csv') as csvfile:
csvReader = csv.reader(csvfile)
for row in csvReader:
items.append(row[0])
print(items)
Example 2: array storing in csv file by python
import numpy as np
import csv
arr1 = [i for i in range(500)]
arr2 = [i for i in range(1000)]
arr3 = [i for i in range(2000)]
with open('record.csv', 'a') as record_append:
np.savetxt(record_append, np.asarray([arr1]), delimiter=',')
np.savetxt(record_append, np.asarray([arr2]), delimiter=',')
np.savetxt(record_append, np.asarray([arr3]), delimiter=',')
two_dim_arr = []
with open('record.csv', 'r') as record_read:
reader = csv.reader(record_read)
for i, each_arr in enumerate(reader):
two_dim_arr.append([eval(each) for each in each_arr])
for each_line in two_dim_arr:
print(each_line)