Python:从CSV文件的每一行提取第一个元素

2024-10-01 04:46:38 发布

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

我想从csv文件中读取并将某些内容添加到列表中

JFK,John F Kennedy International,5326,5486
ORY,Paris-Orly,629,379
MAD,Adolfo Suarez Madrid-Barajas,1428,1151
AMS,Amsterdam Schiphol,526,489
CAI,Cairo International,3779,3584

文本文件中的所有内容都列在上面,我想从每一行中得到第一行,所以JFK CAI添加到了一个列表中

我试过:

with open('Airports.txt', 'r') as f:
    reader = csv.reader(f)
    amr_csv = list(reader)

但这会将整个文本文件添加到列表中,我无法将其仅添加到第一行

总而言之,我想要帮助您将此文本文件的第一行添加到保存为变量的列表中


Tags: 文件csv内容列表johnreader文本文件paris
3条回答

让我们来做一些非常简单的事情

此代码段将IATA代码从CSV文件提取到list

with open('airports.txt') as f:
    iata = [i.split(',')[0] for i in f.readlines()]

代码说明:

本质上,这段代码读取CSV的每一行并用逗号分隔;然后提取第一个元素([0])并使用list comprehension添加到列表中

输出:

['JFK', 'ORY', 'MAD', 'AMS', 'CAI']

我相信这会解决你的问题

import csv
with open('Airports.txt', 'r') as f:
    reader = csv.reader(f)
    amr_csv = list(reader)
    for line in amr_csv:
        print(line[0])

import csv
with open('Airports.txt','r') as f:
    reader = csv.reader(f)
    amr_csv = [line[0] for line in reader]
    print(amr_csv)
import csv

with open('Airports.txt', 'r') as f:
    reader = csv.reader(f)
    example_list = list(reader)

print(example_list)

输出:

[['JFK', 'John F Kennedy International', '5326', '5486'], ['ORY', 'Paris-Orly', '629', '379'], ['MAD', 'Adolfo Suarez Madrid-Barajas', '1428', '1151'], ['AMS', 'Amsterdam Schiphol', '526', '489'], ['CAI', 'Cairo International', '3779', '3584'], []]

感谢所有提供帮助的人,这是我最终决定的,因为这是我一直在寻找的,希望这能帮助任何有类似问题的人

相关问题 更多 >