Reading and writing a CSV file in Breeze is really a breeze. We just have two functions in breeze.linalg package to play with. They are very intuitively named csvread and csvwrite.
In this recipe, you'll see how to:
Read a CSV file into a matrix
Save selected columns of a matrix into a new matrix
Write the newly created matrix into a CSV file
Extract a vector out of the matrix
Write the vector into a CSV
There are just two functions that we need to remember in order to read and write data from and to CSV files. The signatures of the functions are pretty straightforward too:
csvread(file, separator, quote, escape, skipLines) csvwrite(file, mat, separator, quote, escape, skipLines)
Let's look at the parameters by order of importance:
file:java.io.File: Represents the file location.separator: Defaults to a comma so as to represent a CSV. Could be overridden when needed.skipLines: This is the number of lines to be skipped while reading the file. Generally, if there is a header, we pass askipLines=1.mat: While writing, this is the matrix object that is being written.quote: This defaults to double quotes. It is a character that implies that the value inside is one single value.escape: This defaults to a backspace. It is a character used to escape special characters.
Let's see these in action. For the sake of clarity, I have skipped the quote and the escape parameter while calling the csvread and csvwrite functions. For this recipe, we will do three things:
Read a CSV file as a matrix
Extract a sub-matrix out of the read matrix
Write the matrix
Read the CSV as a matrix:
Let's use the
csvreadfunction to read a CSV file into a 100*3 matrix. We'll also skip the header while reading and print 5 rows as a sample:
val usageMatrix=csvread(file=new File("WWWusage.csv"), separator=',', skipLines=1) //print first five rows println ("Usage matrix \n"+ usageMatrix(0 to 5,::)) Output : 1.0 1.0 88.0 2.0 2.0 84.0 3.0 3.0 85.0 4.0 4.0 85.0 5.0 5.0 84.0 6.0 6.0 85.0
Extract a sub-matrix out of the read matrix:
For the sake of generating a submatrix let's skip the first column and save the second and the third column into a new matrix. Let's call it
firstColumnSkipped:val firstColumnSkipped= usageMatrix(::, 1 to usageMatrix.cols-1) //Sample some data so as to ensure we are fine println ("First Column skipped \n"+ firstColumnSkipped(0 to 5, ::)) Output : 1.0 88.0 2.0 84.0 3.0 85.0 4.0 85.0 5.0 84.0 6.0 85.0
Write the matrix:
As a final step, let's write the
firstColumnSkippedmatrix to a new CSV file namedfirstColumnSkipped.csv://Write this modified matrix to a file csvwrite(file=new File ("firstColumnSkipped.csv"), mat=firstColumnSkipped, separator=',')