From ad787929fa092f990effd0486b2d639d430a2627 Mon Sep 17 00:00:00 2001
From: Norbert Walkowiak <norwal2@st.amu.edu.pl>
Date: Sat, 15 Apr 2023 12:22:55 +0200
Subject: [PATCH] add cutoff

---
 s487175-create-dataset-script.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/s487175-create-dataset-script.py b/s487175-create-dataset-script.py
index 10229a9..e107e48 100644
--- a/s487175-create-dataset-script.py
+++ b/s487175-create-dataset-script.py
@@ -1,11 +1,14 @@
 
 #!/usr/bin/env python
-
+import os
 import pandas as pd
 diamonds = pd.read_csv('diamonds.csv')
 #Wyświetlenie zbioru danych
 diamonds
 
+cutoff = int(os.environ['CUTOFF'])
+diamonds = diamonds[:cutoff]
+
 # %%
 #przydzielanie nazwy kolumny z id
 diamonds = diamonds.rename(columns={diamonds.columns[0]: 'id'})