简体   繁体   中英

How to read part of a text file using numpy.loadtxt and convert to array

I have a text file, which has the following content:

type=0
alg=1
c=3 5 7 8 10 11 14 15 16 17 
arr=(0, -7.05154314637184143066e-02) (1, -1.80315405130386352539e-01) (2, -3.32004070281982421875e-01) (3, 5.91193616390228271484e-01) (4, -1.21851079165935516357e-01) (5, -2.71257162094116210938e-01) (6, 2.46545433998107910156e-01) (7, 8.54252055287361145020e-02) 

How can I read from this text file (eg with the name 'file.txt') the array arr and convert it to an array with two columns? The output should be:

array([[ 0.        , -0.07051543],
       [ 1.        , -0.18031541],
       [ 2.        , -0.33200407],
       [ 3.        ,  0.59119362],
       [ 4.        , -0.12185108],
       [ 5.        , -0.27125716],
       [ 6.        ,  0.24654543],
       [ 7.        ,  0.08542521]])

If I would have this line:

0,-7.05154314637184143066e-02,1,-1.80315405130386352539e-01,2,-3.32004070281982421875e-01,3, 5.91193616390228271484e-01,4,-1.21851079165935516357e-01,5,-2.71257162094116210938e-01,6, 2.46545433998107910156e-01,7,8.54252055287361145020e-02 

I could use:

import numpy as np
arr=np.loadtxt('file.txt', delimiter=',')
arr.reshape(-1,2)

Simple Python will get you to where you can use np.fromstring :

for line in open('file.txt'):
    if line.startswith('arr='):
        line = line[5:].rstrip().rstrip(')').replace(') (', ',')
        arr = np.fromstring(line, sep=',').reshape(-1, 2)
        break  # or return arr
else:
    raise ValueError('SOL!')

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM