audit-labs/audit-tools

A collection of scripts, queries, and other goodies you can use in an audit.

clone: git clone https://gitbay.org/audit-labs/audit-tools.git

9bf4bc59ee673fb72e56104aec94282ec55bb7ee

verified · cmc

author: Christian Cleberg <hello@cleberg.net> · 2024-10-19T16:26:23Z

add sample.py

 sample.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/sample.py b/sample.py
new file mode 100644
index 0000000..a2435d8
--- /dev/null
+++ b/sample.py
@@ -0,0 +1,30 @@
+# Import packages
+import pandas as pd
+
+# Define the sample size
+sample_size = 25
+
+# Import the data to a pandas DataFrame
+df = pd.read_csv("FILENAME_GOES_HERE.csv")
+
+# ALTERNATIVE: If you use Excel, use this instead. Supports xls, xlsx, xlsm,
+# xlsb, odf, ods and odt file extensions.
+# df = pd.read_excel("FILENAME_GOES_HERE.xlsx")
+
+# Print totals prior to sampling
+print("Dataframe size (rows, columns): ", df.shape)
+
+# Sample
+sample = df.sample(sample_size)
+print("Sample size: ", sample_size)
+print("Sample:\n", sample)
+
+# ALTERNATIVE: Replacement Samples
+#
+# If you want replacement samples (e.g., 10 samples & 3 replacements), you will
+# need to increase sample size to the total you want (e.g., 13). If that is
+# larger than the population, you will need to use the `replace=True` parameter.
+#
+# # Sample Size: 25 + 5 replacement samples
+# sample_size = 30
+# sample = df.sample(30, replace=True)