saving numpy array to file with a certain precision

Question

I have a following numpy array:

geom= [[ 6.          0.2        -1.6        -1.3915    ]
         [ 6.          1.40507435 -1.6        -0.69575   ]
         [ 6.          1.40507435 -1.6         0.69575   ]
         [ 6.          0.2        -1.6         1.3915    ]
         [ 6.         -1.00507435 -1.6         0.69575   ]
         [ 6.         -1.00507435 -1.6        -0.69575   ]]

When I save it to the file :

np.savetxt(g, geom, fmt ='%f6', delimiter=' ', newline='\n', header='', footer='', comments='# ')

My first column gets a format '6.0000008'. Subsequently I want to modify that file to obtain 6.0 or 6. Is there any easy solution to do it? I have tried to convert string to int, but I am getting the error message in return:

ValueError: invalid literal for int() with base 10: '6.0000008'

You might try using pandas instead of numpy. It lets you easily combine data of different types into a single dataframe, which can then be exported to file in a much cleaner way. — jakevdp
– jakevdp, Commented Apr 10, 2017 at 15:13

Grr · Accepted Answer · 2017-04-10 15:24:39Z

1

Assuming you would like to keep the other floats as is you aren't going to find a simple solution to this in NumPy. Granted you could probably use structured arrays to achieve what you want but it will be significantly more work than just using Pandas (as mentioned by @jakevdp).

You could just import your NumPy array into a Pandas DataFrame, make the changes you want and then output the data into any Pandas acceptable format.

For example, to write a csv:

df = pd.DataFrame(geom)
df[0] = df[0].astype(int)
df.to_csv(g, index=False)

answered Apr 10, 2017 at 15:24

Grr

16.2k7 gold badges72 silver badges91 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

saving numpy array to file with a certain precision

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related