Kuramoto/CSV_vectorReader.py at master · AndrewJSON/Kuramoto · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

'''
 * CSV_vectorReader.py
 *
 *   Created on:         15.02.2019
 *   Author:             Andrew Jason Bishop
 *
 * General description:
 *   https://realpython.com/python-csv/
'''

import csv
import numpy as np
from itertools import islice


class CSV_vectorReader:

    def __init__(self):
        self.vectorLength = 1


    def setVectorLength(self, _N):
        self.vectorLength = _N


    def getVectorBlockFromFile(self, _fileName, _blockSize):

        vectorBlock = self.getPreparedNumpyArray( _blockSize )

        with open( _fileName, mode='r' ) as in_file:
            reader = csv.reader( in_file                  ,\
                                 delimiter=','            ,\
                                 quoting=csv.QUOTE_MINIMAL )

            for i in range(0, _blockSize):

                row = next(reader)
                npRow = np.asarray(row)
                npRow = npRow.astype(np.float)

                vectorBlock[i] = npRow

            print("successfully read", _fileName, )

        return vectorBlock


    def getPreparedNumpyArray(self, _blockSize):
        return np.zeros( (_blockSize, self.vectorLength) )


    def getVectorBlock(self, _start, _stop, _fileName='phase-results.csv' ):

        blockSize = _stop - _start
        vectorBlock = self.getPreparedNumpyArray( blockSize )

        with open( _fileName, mode='r' ) as in_file:

            reader = csv.reader( in_file, delimiter=',' )
            block  = islice(reader, _start, _stop)

            for i, row in enumerate(block):

                npRow = np.asarray(row)
                npRow = npRow.astype(np.float)
                vectorBlock[i] = npRow

        return vectorBlock


    def printVectorBlock(self, _start, _stop, _fileName='phase-results.csv' ):

        blockSize = _stop - _start
        vectorBlock = self.getPreparedNumpyArray( blockSize )

        with open( _fileName, mode='r' ) as in_file:

            reader = csv.reader( in_file, delimiter=',' )
            block  = islice(reader, _start, _stop)

            for i, row in enumerate(block):

                npRow = np.asarray(row)
                npRow = npRow.astype(np.float)

                print(i, npRow)


'''
        with open('my_data.csv') as fd:
            for row in islice(csv.reader(fd), _row, None):
                print(row)
'''

''' END '''