argparse_example.py

import argparse
import pandas as pd

if __name__ == "__main__":
  
  # create argument parser and define arguments
  parser = argparse.ArgumentParser()
  parser.add_argument("--inputfile", default="inputfile.csv")
  parser.add_argument("--num_rows", type=int, default=10)
  parser.add_argument("--print_output", action="store_true")
  args = parser.parse_args()
  
  # read in dataset
  df = pd.read_csv(args.inputfile)
  df_subset = df.iloc[:args.num_rows, :]
  
  # print dataset if flag is set
  if args.print_output:
    print(df_subset)