csv2npz.py

import pandas as pd
import numpy as np
import sys

filename = sys.argv[1]


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


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


if __name__ == "__main__":
    main()