擅长:python、mysql、java
<p>不确定<code>head</code>和头逻辑,但可以使用<code>itertools.islice</code>拉取第一行<code>header_length</code>和<code>str.join</code>来连接输出:</p>
<pre><code>from itertools import islice
filename = raw_input("Which file are we loading? "))
# ask user how many header lines
header_length = int(raw_input("Enter amount of header lines"))
with open(filename) as myfile:
# get the first header_length lines in a list
head = list(islice(myfile, header_length))
header = raw_input("How many header lines? (Type ``head`` to see the header lines)")
# if user types head
if "head" == header:
# use enumerate to get the line numbers/index in list
# the str.join the lines formatting index | line
print("".join(["{} | {}".format(i, line) for i, line in enumerate(head,start=1)]))
</code></pre>