我怎样才能将某些输入只读到我的列表中?

时间:2018-04-24 08:11:23

标签: python csv

我的数据如下:

image       id      name    xMin    xMax    yMin    yMax
24-5.png    1               4632    4695      42    57
24-5.png    2               2910    2962      60    74
24-5.png    3               2976    3045      60    74
24-5.png    4               2902    2980      84    99
45-11.png   1463            1209    1240    3455    3469
45-11.png   1464            1246    1300    3459    3470

在那里可以看到'有一个.csv文件,让我们称之为data.csv我怎样才能用24-5.png读取图像以进一步处理? 我目前正在阅读的方式是

labels1 = []
with open("data.csv", 'r') as f:
    reader = csv.DictReader(f, delimiter='\t')
    for line in reader:
        labels1.append(line)

基本上我希望label1包含相同的数据格式,但仅限于图像的特定值。

1 个答案:

答案 0 :(得分:0)

labels1 = []
with open("data.csv", 'r') as f:
    reader = csv.DictReader(f, delimiter='\t')
    for line in reader:
        if line["image"] == "24-5.png":     #Add an if to check for value of "image"
            labels1.append(line)