add cutoff

2023-04-15 12:22:55 +02:00 · 2023-04-15 12:22:55 +02:00 · ad787929fa
commit ad787929fa
parent ead7af856f
1 changed files with 4 additions and 1 deletions
--- 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'})