从文件

时间:2018-06-14 23:25:53

标签: python python-3.x

我从文件中读到:

with open(fileName) as f:
    for line in f:
        print(line.split(","))     #split the file into multiple lists

如何从这些列表中获取某些特定元素?

例如,只有索引为[0到3]的元素,但在此之后丢弃/忽略任何元素。

2 个答案:

答案 0 :(得分:1)

请尝试以下代码:

with open('f.txt') as f:
    print('\n'.join([i for i in f.read().split(',')[0:3]]))

答案 1 :(得分:1)

如果要保存每行中的前三项,可以使用列表推导

with open(fileName) as f:
    firstitems = [line.rstrip().split(",")[0:3] for line in f]

请注意,如果一行中的项目少于四个,则需要rstrip()来删除最终的换行符。请注意"项目"都是字符串,即使它们看起来像其他类型。例如,如果你想要整数,你需要将它们转换为整数。

然后你可以打印出来:

for line in firstitems:
    print(line)