From 03260973785249f8e36bd697b2c129657b2adbe0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Wojciech=20Miko=C5=82ajski?= <wojmik2@st.amu.edu.pl>
Date: Mon, 17 Apr 2023 21:28:02 +0200
Subject: [PATCH] Update 'create_dataset.py'

---
 create_dataset.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/create_dataset.py b/create_dataset.py
index 30d17e0..66630ab 100644
--- a/create_dataset.py
+++ b/create_dataset.py
@@ -5,9 +5,13 @@
 #api.authenticate()
 #api.dataset_download_file('mssmartypants/water-quality', file_name='waterQuality1.csv')
 
+import argparse
+parser.add_argument('CUT', type=int, nargs='?', default=7999, help='Cut this number of rows from csv file')
+
 #Reading DataSet from csv using pandas library
 import pandas as pd
-water = pd.read_csv('waterQuality1.csv')
+water = pd.read_csv(os.path.join('.', 'waterQuality1.csv'), nrows = args.CUT)
+
 
 #water.describe(include='all')