在python中迭代txt中的第一个元素

Question

我有一个txt文件，我只想为前3行中每行的元素数量进行计数。

我尝试这样做：

for line in open('C:\Users\Gabriela\Documents\prueba_py.txt')[:3]: 
    arr = line.strip().split('\t') 
    print len(arr) 

但不起作用。我怎样才能做到这一点？

谢谢！

Answer 1

文件句柄不支持切片[:3]。也许你想要open().readlines()[:3]？要注意的是加载到内存整个文件......我会用这个来代替：

for lineno, line in enumerate(open(...)): 
    if lineno > 2: 
    break 
    # Remaining code here.