/Python/Yield value from csv file python

Yield value from csv file python

import csv
import numpy as np

def getData(filename1, filename2):
    with open(filename1, "rb") as csv1, open(filename2, "rb") as csv2:
        reader1 = csv.reader(csv1)
        reader2 = csv.reader(csv2)
        for row1, row2 in zip(reader1, reader2):
            yield (np.array(row1, dtype=np.float),
                   np.array(row2, dtype=np.float)) 
                # This will give arrays of floats, for other types change dtype

for tup in getData("file1", "file2"):
    print(tup)

Source: stackoverflow.com

Similar codes

0 vote104 views1 answer
Choose custom index pandas
Python
0 vote118 views1 answer
Take the first in dataloader pytorch
Python
3 votes100 views1 answer
Pandas reciprocal
Python
23 votes128 views4 answers
Pandas show all rows
Python
0 vote92 views1 answer
Pandas dataframe unique multiple columns
Python
13 votes110 views5 answers
How to check if a input is an integer python
Python
17 votes64 views1 answer
Strip all spaces python
Python
0 vote104 views1 answer
Python count distinct letters
Python
0 vote106 views1 answer
When training= false still dropout
Python
2 votes113 views1 answer
Create an array string using for in python
Python