csv2npy.py

import pandas as pd
import numpy as np
import sys

# for f in rumi*.csv; do python csv2npy.py $f; done

filename = sys.argv[1]


def csv2npy(filename):
    df = pd.read_csv(filename)
    data = df.values
    np.save(filename[:-4], data)


def main():
    if filename.endswith(".csv"):
        csv2npy(filename)
    else:
        print("Please provide a csv file")


if __name__ == "__main__":
    main()