earthjasonlin/GachaSimulator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

specify config file

Browse Source
This commit is contained in:
Zichao Lin 2023-09-02 18:24:14 +08:00
parent 83ffa41771
commit 20ce50ff24
Signed by: earthjasonlin
GPG Key ID: 406D9913DE2E42FB
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
main.py
View File

@ -2,7 +2,8 @@ import json
import random import random
import csv import csv
from tqdm import tqdm from tqdm import tqdm
import argparse
import os
class ProbabilityTable: class ProbabilityTable:
def __init__(self, table): def __init__(self, table):
@ -54,7 +55,7 @@ class Statistics:
results[n]["samples"].append(i + 1) results[n]["samples"].append(i + 1)
pbar.update() pbar.update()
with open("results_full.csv", "w", newline="") as f: with open(results_full_filename, "w", newline="") as f:
writer = csv.writer(f) writer = csv.writer(f)
writer.writerow(["抽数", "该抽出金的概率", "刚好在该抽出金的概率", "该抽出金的样本数量", "该抽出金的样本号"]) writer.writerow(["抽数", "该抽出金的概率", "刚好在该抽出金的概率", "该抽出金的样本数量", "该抽出金的样本号"])
for n in range(1, maximum_draws + 1): for n in range(1, maximum_draws + 1):
@ -65,7 +66,7 @@ class Statistics:
[n, p, probability_of_exact_n, results[n]["count"], samples_str] [n, p, probability_of_exact_n, results[n]["count"], samples_str]
) )
with open("results_no_samples.csv", "w", newline="") as f_no_samples: with open(results_no_samples_filename, "w", newline="") as f_no_samples:
writer_no_samples = csv.writer(f_no_samples) writer_no_samples = csv.writer(f_no_samples)
writer_no_samples.writerow(["抽数", "该抽出金的概率", "刚好在该抽出金的概率", "该抽出金的样本数量"]) writer_no_samples.writerow(["抽数", "该抽出金的概率", "刚好在该抽出金的概率", "该抽出金的样本数量"])
for n in range(1, maximum_draws + 1): for n in range(1, maximum_draws + 1):
@ -76,11 +77,20 @@ class Statistics:
) )
with open("config.json", "r") as f: parser = argparse.ArgumentParser(description="Take user defined parameters.")
parser.add_argument("--config", "-c", type=str, required=True, help="Path to your config.json file")
args = parser.parse_args()
with open(args.config, "r") as f:
config = json.load(f) config = json.load(f)
table = ProbabilityTable(config["probability_table"]) table = ProbabilityTable(config["probability_table"])
game = CardGame(table) game = CardGame(table)
statistics = Statistics(game, config["num_trials"]) statistics = Statistics(game, config["num_trials"])
config_name, _ = os.path.splitext(args.config)
results_full_filename = f"results_{config_name}_full.csv"
results_no_samples_filename = f"results_{config_name}_no_samples.csv"
statistics.run() statistics.run()