python - 如何获取行中的特定列(csv)

Question

我有这段代码:

with open(filepath, 'rb') as csvfile:
    reader = csv.reader(csvfile, delimiter=' ')
    for row in reader:
        print(' '.join(row))

它返回:

Type,UniProt ID,Position
Sodium channel,P35498,1-123
Sodium channel,P35498,176-188
Sodium channel,P35498,234-239
Sodium channel,P35498,426-762
Sodium channel,P35498,823-830
Potassium channel ATP-sensitive inward rectifier,P48048,1-77

我希望能够将 P35498 放入字符串中以供稍后使用。我该怎么做？我也希望能够获得这些列中的任何一个，但只要有 P35498 的示例就太好了，谢谢!

如果我这样做

with open(filepath, 'rb') as csvfile:
    reader = csv.reader(csvfile, delimiter=',')
    idlist = []
    for row in reader:
        idlist.append(row[1])
        print(idlist)

返回:

['UniProt ID']
['UniProt ID', 'P35498']
['UniProt ID', 'P35498', 'P35498']
['UniProt ID', 'P35498', 'P35498', 'P35498']
['UniProt ID', 'P35498', 'P35498', 'P35498', 'P35498']
['UniProt ID', 'P35498', 'P35498', 'P35498', 'P35498', 'P35498']
['UniProt ID', 'P35498', 'P35498', 'P35498', 'P35498', 'P35498', 'P48048']

Answer 1

如果您想要第 2 列中有“P35498”的行，您可以检查第 2 列并在相同时打印:

with open(filepath, 'rb') as csvfile:
    reader = csv.reader(csvfile, delimiter=' ')
    for row in reader:
        if "P35498"==row[1]:
            print(' '.join(row))