python ターミナルからファイルの入力

Question

python3.6でファイルを引数として受け取り処理するプログラムを作成しているのですが、jupyter notebookでうまく動いたファイルを.pyファイルとしてマックのターミナルから動かそうとするとエラーが出ます。
原因わかりますでしょうか。

import sys
import pandas as pd
from datetime import datetime as dt

args = sys.argv

def main(args):
        df = pd.read_csv(args)
        tstr = str(df.iat[0,0])
        print(tstr)
        tdatetime = dt.strptime(tstr,'%H:%M')

        if tdatetime.hour -12 > 0:
                return True
        else:
                return False



main(args)

ターミナルからの入力
python3.6 test.py test.csv

test.csvの内容

date,col1
16:49,aaa

エラー内容：
ValueError: Invalid file path or buffer object type: <class 'list'>

よろしくお願いいたします。

Answer 1

sys.argvはコマンドラインに指定した引数(文字列)のリストです。
そのままread_csvの引数にファイル名として指定することはできません。

提示されたmain関数がひとつのファイルを扱うように書かれているので、sys.argvから、(自分自身を除いた)要素をひとつずつ取り出してmainに渡すといいのでは。

import sys
import pandas as pd
from datetime import datetime as dt

def main(file):
    df = pd.read_csv(file)
    tstr = str(df.iat[0, 0])
    print(tstr)
    tdatetime = dt.strptime(tstr, '%H:%M')

    if tdatetime.hour - 12 > 0:
        return True
    else:
        return False

for f in sys.argv[1:]:
    main(f)

Comments

[yuyu127]

Daregadaさん
早急に教えていただきありがとうございました。
sys.argvは引数のリストを渡すということ、大変勉強になりました。
うまくスクリプトを動かすことができました。ありがとうございました。