I know that I can load the datasets in the pandas using the read_csv function of pandas as below
import pandas as pd
df = read_csv('/home/user/iris_dataset.csv', header=0)
df.head()
How can I load the same dataset using the command line arguments while executing the code ? I want to load the dataset to a data frame like python3 example.py -D 'iris_dataset.csv' while executing the code
sys.argvWe can use sys.argv like so :
import pandas as pd
import sys
df = pd.read_csv(f'/home/user/{sys.argv[1]}', header=0)
print(df.head())
We can call it using :
python example.py iris_dataset.csv
argparseOr we can use argparse :
import argparse
import pandas as pd
my_parser = argparse.ArgumentParser()
my_parser.add_argument('-D', action='store', type=str, required=True)
args = my_parser.parse_args()
df = pd.read_csv(f'/home/user/{args.D}', header=0)
print(df.head())
It works correctly using :
python example.py -D "iris_dataset.csv"
int or a float, you can set up type=float in add_argument as int is a subset of float. Or you can just remove the type parameter to accept whatever type you pass to it.