一般的なカスタム関数の概要

5284 ワード

ファイルパスの取得
def get_file_path(rootdir):
    for (dirpath, dirnames, filenames) in os.walk(rootdir):
        pathname = [os.path.join(dirpath, filename) for filename in filenames]
    return pathname

日付回転タイムスタンプ
def time2stamp(timestr):
    timeArray = time.strptime(timestr, "%Y-%m-%d %H:%M")
    # year = timeArray.tm_year
    timeStamp = int(time.mktime(timeArray))
    return timeStamp

タイムスタンプ日付(pandasがデータを処理する場合、値がNULLの場合はNaN)
def stamp2time(timestamp, dt=''):  #         
    if not timestamp is np.NaN:
        dt = time.strftime("%Y-%m-%d %H:%M", time.localtime(int(timestamp)))
    return dt

pandasデータベースへの書き込み
def write2sql(df, tableName):
    con = create_engine('mysql+pymysql://user:password@host:3306/database?charset=utf8')
    df.index = range(1, len(df) + 1)
    df.to_sql(tableName, con, schema='database', if_exists='append', index=True, index_label='id')  #     chunksize=10000;   :index false