Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to write Fortran Output as CSV file?

Can any one tell me, how can I write my output of Fortran program in CSV format? So I can open the CSV file in Excel for plotting data.

like image 460
aibk01 Avatar asked Aug 20 '11 07:08

aibk01


4 Answers

I thought a full simple example without any other library might help. I assume you are working with matrices, since you want to plot from Excel (in any case it should be easy to extend the example).

tl;dr

Print one row at a time in a loop using the format format(1x, *(g0, ", "))

Full story

The purpose of the code below is to write in CSV format (that you can easily import in Excel) a (3x4) matrix. The important line is the one labeled 101. It sets the format.

program testcsv
   IMPLICIT NONE 

   INTEGER :: i, nrow
   REAL, DIMENSION(3,4) :: matrix

   ! Create a sample matrix
   matrix = RESHAPE(source = (/1,2,3,4,5,6,7,8,9,10,11,12/), &
                    shape = (/ 3, 4 /))

   ! Store the number of rows
   nrow = SIZE(matrix, 1) 

   ! Formatting for CSV
   101 format(1x, *(g0, ", ")) 
   
   ! Open connection (i.e. create file where to write)
   OPEN(unit = 10, access = "sequential", action = "write", &
        status = "replace", file = "data.csv", form = "formatted") 
   ! Loop across rows
   do i=1,3
      WRITE(10, 101) matrix(i,:)
   end do 
   ! Close connection
   CLOSE(10)


end program testcsv

We first create the sample matrix. Then store the number of rows in the variable nrow (this is useful when you are not sure of the matrix's dimension beforehand). Skip a second the format statement. What we do next is to open (create or replace) the CSV file, names data.csv. Then we loop over the rows (do statement) of the matrix to write a row at a time (write statement) in the CSV file; rows will be appended one after another.

In more details how the write statement works is: WRITE(U,FMT) WHAT. We write "what" (the i-th row of the matrix: matrix(i,:)), to connection U (the one we created with the open statement), formatting the WHAT according to FMT.

Note that in the example FMT=101, and 101 is the label of our format statement:

format(1x, *(g0, ", ")) 

what this does is: "1x" insert a white space at the beginning of the row; the "*" is used for unlimited format repetition, which means that the format in the following parentheses is repeated for all the data left in the object we are printing (i.e. all elements in the matrix's row). Thus, each row number is formatted as: 'g0, ", "'. g is a general format descriptor that handles floats as well as characters, logicals and integers; the trailing 0 basically means: "use the least amount of space needed to contain the object to be formatted" (avoids unnecessary spaces). Then, after the formatted number, we require the comma plus a space: **", ". This produces our comma-separated values for a row of the matrix (you can use other separators instead of "," if you need). We repeat for every row and that's it.

(The spaces in the format are not really needed, thus one could use format(*(g0,","))

Reference: Metcalf, M., Reid, J., & Cohen, M. (2018). Modern Fortran Explained: Incorporating Fortran 2018. Oxford University Press.

like image 80
luco00 Avatar answered Nov 07 '22 14:11

luco00


I'd also recommend the csv_file module from FLIBS. Fortran is well equipped to read csv files, but not so much to write them. With the csv_file module, you put

    use csv_file

at the beginning of your function/subroutine and then call it with:

    call csv_write(unit, value, advance)

where unit = the file unit number, value = the array or scalar value you want to write, and advance = .true. or .false. depending on whether you want to advance to the next line or not.

Sample program:

  program write_csv

    use csv_file

    implicit none

    integer :: a(3), b(2)

    open(unit=1,file='test.txt',status='unknown')

    a = (/1,2,3/)
    b = (/4,5/)

    call csv_write(1,a,.true.)
    call csv_write(1,b,.true.)

  end program

output:

1,2,3

4,5

if you instead just want to use the write command, I think you have to do it like this:

    write(1,'(I1,A,I1,A,I1)') a(1),',',a(2),',',a(3)
    write(1,'(I1,A,I1)') b(1),',',b(2)

which is very convoluted and requires you to know the maximum number of digits your values will have.

I'd strongly suggest using the csv_file module. It's certainly saved me many hours of frustration.

like image 21
bananafish Avatar answered Nov 07 '22 12:11

bananafish


A slightly simpler version of the write statement could be:

write (1, '(1x, F, 3(",", F))') a(1), a(2), a(3), a(4)

Of course, this only works if your data is numeric or easily repeatable. You can leave the formatting to your spreadsheet program or be more explicit here.

like image 11
Mike Avatar answered Nov 07 '22 13:11

Mike


The Intel and gfortran (5.5) compilers recognize:

write(unit,'(*(G0.6,:,","))')array or data structure

which doesn't have excess blanks, and the line can have more than 999 columns.

To remove excess blanks with F95, first write into a character buffer and then use your own CSV_write program to take out the excess blanks, like this:

write(Buf,'(999(G21.6,:,","))')array or data structure
call CSV_write(unit,Buf)

You can also use

write(Buf,*)array or data structure
call CSV_write(unit,Buf)

where your CSV_write program replaces whitespace with "," in Buf. This is problematic in that it doesn't separate character variables unless there are extra blanks (i.e. 'a ','abc ' is OK).

like image 5
user3772612 Avatar answered Nov 07 '22 12:11

user3772612