自然数Nをコマンドライン引数などの手段で受け取り,入力のうち先頭のN行だけを表示せよ.というものをPythonで書いたのですが、とりあえず書けただけ感が否めないコードになってしまいました。もう少しPythonらしい書き方ができないでしょうか。
import sys
i = 0
with open("hightemp.txt", "r") as f:
whole_str = f.readlines()
while(i != int(sys.argv[1])):
print(whole_str[i])
i += 1