I'm trying to turn a csv file into arrays.testdata here
what I have in csv file is like this
11,10,8,12,13,11
0,1,0,2,3,0
5,15,13,11,18,18
I want to turn it into arrays as below,
[[[11],
[10],
[8],
[12],
[13],
[11]],
[[0],
[1],
[0],
[2],
[3],
[0]],
[[5],
[15],
[13],
[11],
[18],
[18]]]
Read the file and get the items as list from it:
import csv
results = []
with open('some_array.csv','r') as f:
lines = csv.reader(f)
for line in lines:
results.append([[int(i)] for i in line])
>>results
[[['11'], ['10'], ['8'], ['12'], ['13'], ['11']],
[['0'], ['1'], ['0'], ['2'], ['3'], ['0']],
[['5'], ['15'], ['13'], ['11'], ['18'], ['18']]]
Solution:
import csv
with open('input.csv', 'rb') as f:
reader = csv.reader(f)
lines = [row for row in reader]
array = [map(int,l) for l in lines]
array
[[11, 10, 8, 12, 13, 11], [0, 1, 0, 2, 3, 0], [5, 15, 13, 11, 18, 18]]
import csv
output = []
with open('test1.csv', 'r') as f:
content = csv.reader(f, delimiter=',')
for line in content:
clean_line = filter(None, line) # remove extra spaces
output.append([[int(i)] for i in clean_line])
>>> print output
[[[11], [10], [8], [12], [13], [11]], [[0], [1], [0], [2], [3], [0]], [[5], [15], [13], [11], [18], [18]]]
Tested and results match desired output:
desired = [ [[11],[10],[8],[12],[13],[11]],
[[0],[1],[0],[2],[3],[0]],
[[5],[15],[13],[11],[18],[18]] ]
# print output == desired # True
