如何将特定文本保存在 .txt 文件中的数组中

Question

I have a .txt file, where I want to save only following characters "N", "1.1" ,"XY", "N", "2.3" ,"xz" in an array. The .txt file looks like this:

[   TITLE

    N 1.1 XY
    N 2.3 XZ

]

Here is my code:

src = open("In.txt", "r")

def findOp (row):
    trig = False
    temp = ["", "", ""]
    i = 1
    n = 0
    for char in row:  
        i += 1
        if (char != '\t') & (char != ' ') & (char != '\n'):
            trig = True
            temp[n] += char
        else:
            if trig:
                n += 1
                trig = False

    return temp

for line in src.readlines():
print(findOp(line))

The Output from my code is:

['[', 'TITLE', '']
['', '', '']
['N', '1.1', 'XY']
['N', '2.3', 'XZ']
['', '', '']
[']', '', '']

The problem is the program also saves whitespace characters in an array which i dont want.

Answer 1

I would recommend the trim()-function with witch one you can remove whitespace from a string

Whitespace on both sides:

s = s.strip()

Whitespace on the right side:

s = s.rstrip()

Whitespace on the left side:

s = s.lstrip()

Answer 2

Try this :

with open('In.txt', 'r') as f:
    lines = [i.strip() for i in f.readlines() if i.strip()][1:-1]
output = [[word for word in line.split() if word] for line in lines]

Output :

[['N', '1.1', 'XY'], ['N', '2.3', 'XZ']]

Answer 3

Try numpy.genfromtxt :

import numpy as np
text_arr = np.genfromtxt('In.txt', skip_header = 1, skip_footer = 1, dtype = str)
print(text_arr)

Output:

[['N' '1.1' 'XY']
 ['N' '2.3' 'XZ']]

Or if you want list, add text_arr.tolist()

Answer 4

You could check the return array before exiting:

def findOp(row):
    trig = False
    temp = ["", "", ""]
    i = 1
    n = 0
    for char in row:
        i += 1
        if (char != '\t') & (char != ' ') & (char != '\n'):
            trig = True
            temp[n] += char
        else:
            if trig:
                n += 1
                trig = False

    # Will return `temp` if all elements eval to True otherwise
    # it will return None        
    return temp if all(temp) else None

The value None can then be used as a check condition in subsequent constructs:

for line in src.readlines():
    out = findOp(line)
    if out:
        print(out)

>> ['N', '1.1', 'XY']
>> ['N', '2.3', 'XZ']