入力の先頭を表示するPythonらしいプログラムの書き方


自然数Nをコマンドライン引数などの手段で受け取り，入力のうち先頭のN行だけを表示せよ.というものをPythonで書いたのですが、とりあえず書けただけ感が否めないコードになってしまいました。もう少しPythonらしい書き方ができないでしょうか。

[code]

    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
    
[/code]




可読性などに問題がなければあまり細かいこと気にしなくてもいいと思いますが、以下のようにも書けます:

[code]

    import sys
    
    with open("hightemp.txt") as f:
        for i in range(int(sys.argv[1])):
            print(f.readline())
    
[/code]

