一次迭代一行二维numpy数组python

Question

我有如下所示的数组。

[['dad', 'mom', 'test1.txt']
['brother', 'sister', 'test2.txt']
['uncle', 'aunty', 'test3.txt']
['grandpa', 'grandma', 'test4.txt']]

例如，我想迭代。我从 第一个第一个元素 打开 test1.txt 文件，如果字符串 'dad' 和 'mom' 出现在拆分索引中，则过滤文件并读取并附加年龄.同样对于 text2.txt 文件，如果存在 'brother' 和 'sister' 然后读取并过滤，我希望它在嵌套列表中不一样。

我有类似下面的东西

lst_age=[]
for data in ele_arry:
    file = data[2]
    stg1 = data[1]
    stg2 = data[0]
    files = open(file, 'r').read()
    for line in files:
        formattedline = line.split(' ')
        if formattedline[1] == stg1 and formattedline[2] == stg2:
            lst_age.append(formattedline[3])

逐行分割后的文本文件如下：

['1','dad', 'mom', '27']
['2','dad', 'mom', '34']
['3','daughter', 'mom', '42']
['4','dad', 'son', '21']
['5','daughter', 'son', '22']

我需要的输出是 lst_age 变量

['27', '34'] #for file text1.txt
['28', '43' ] #for file text2.txt
#similarly for all the 4 files 

但现在我将所有文件中的年龄元素合并为一个。

['27', '34','28', '43'] 

这没有提供上述所需的输出。有人可以帮我解决这个问题吗？

谢谢！

Answer 1

这是因为使用files = open(file, 'r').read() 你会得到一个文件对象而不是一个可迭代的列表。您可以使用 readlines() 返回包含文件行的列表：

ele_arry =[['dad', 'mom', 'C:\\Users\\Miguel\\Desktop\\test1.txt'],
['brother', 'sister', 'C:\\Users\\Miguel\\Desktop\\test2.txt'],
['uncle', 'aunty', 'C:\\Users\\Miguel\\Desktop\\test3.txt'],
['grandpa', 'grandma', 'C:\\Users\\Miguel\\Desktop\\test4.txt']]


lst_age=[]
for data in ele_arry:
    file = data[2]
    stg2 = data[1]
    stg1 = data[0]
    files = open(file, 'r').readlines()
    lst_agefile=[]
    for line in files:
        formattedline = line.split(' ')
        if formattedline[1] == stg1 and formattedline[2] == stg2:
            lst_agefile.append(formattedline[3].replace('\n',''))
    lst_age.append(lst_agefile)
for i in lst_age:
    print(i)

输出：

['27', '34'] 
['28', '43' ]

单行方式：

lst_age=[[line.split(' ')[3].replace('\n','') for line in open(data[2], 'r').readlines() if line.split(' ')[1] == data[0] and line.split(' ')[2] ==  data[1]] for data in ele_arry]