如何在python中垂直读取列表

2024-06-01 19:49:23 发布

您现在位置:Python中文网/ 问答频道 /正文

我的代码有点纠结。所以我有一个.txt文件要读,并将其列在列表中。这是我的.txt文件,它已经进入列表

['0--Parade/0_Parade_marchingband_1_849.jpg']
['1']
['449', '330', '122', '149', '0', '0', '0', '0', '0', '0', '']
['0--Parade/0_Parade_Parade_0_904.jpg']
['1']
['361', '98', '263', '339', '0', '0', '0', '0', '0', '0', '']

这是我读取.txt文件的代码

import numpy as np
import os

def extract_and_filter_data(splits):
    bb_gt_collection = dict()

    for split in splits:
        with open(
                os.path.join('dataset', 'wider_face_split',
                             '%s.txt' % (split)), 'r') as f:
            lines = f.readlines()

        for line in lines:
            line = line.split('\n')[0]
            if line.endswith('.jpg'):
                image_path = os.path.join('dataset', 'WIDER_%s' % (split),
                                          'images', line)
                bb_gt_collection[image_path] = []



            line_components = line.split(' ')
            print(line_components)

所以,我想从这些列表中得到数字1。但我不知道如何垂直获得该值。有人能帮我吗?如果这是个愚蠢的问题,我很抱歉。任何答复都将不胜感激。谢谢


Tags: 文件path代码importgttxt列表os