如何根据Python中的字符串值从列表中选择某些值？

Question

我有一个超链接列表，其中包含三种类型的链接。 htm ， csv和pdf 。 我只想挑出csv那些。

该列表包含以下格式的字符串： csv/damlbmp/20160701damlbmp_zone_csv.zip

我当时想在字符串上运行一个for循环，只返回前三个字符串值等于csv值，但是我不确定如何做到这一点。

Answer 1

如果您的列表称为links ：

[x for x in links if 'csv/' in x]

Answer 2

我会使用link.endswith('csv') （或link.endswith('csv.zip') ），其中link是包含该链接的字符串）

例如：

lst = ['csv/damlbmp/20160701damlbmp_zone_csv.zip',
       'pdf/damlbmp/20160701damlbmp_zone_pdf.zip',
       'html/damlbmp/20160701damlbmp_zone_html.zip',
       'csv/damlbmp/20160801damlbmp_zone_csv.zip']

csv_files = [link for link in lst if link.endswith('csv.zip')]

Answer 3

你可以试试这个

import re
l=["www.h.com","abc.csv","test.pdf","another.csv"] #list of links

def MatchCSV(list):
    matches=[]
    for string in list:
        m=re.findall('[^\.]*\.csv',string)
        if(len(m)>0):
            matches.append(m)
    return matches

print(MatchCSV(l))
[['abc.csv'], ['another.csv']]

（endswith也是一个不错的选择）

Answer 4

这是一种方法：

lst = ['csv/damlbmp/20160701damlbmp_zone_csv.zip',
       'pdf/damlbmp/20160701damlbmp_zone_pdf.zip',
       'html/damlbmp/20160701damlbmp_zone_html.zip',
       'csv/damlbmp/20160801damlbmp_zone_csv.zip']

[i for i in lst if i[:3]=='csv']

# ['csv/damlbmp/20160701damlbmp_zone_csv.zip',
#  'csv/damlbmp/20160801damlbmp_zone_csv.zip']