Compare commits
No commits in common. "main" and "tut5-v1.2" have entirely different histories.
@ -1,38 +0,0 @@
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plt
|
||||
from numpy.typing import NDArray
|
||||
import argparse
|
||||
|
||||
|
||||
def twodim_array_to_pgfplots_table_string(a: NDArray):
|
||||
return (
|
||||
" \\\\\n".join([" ".join([str(vali) for vali in val]) for val in a]) + "\\\\\n"
|
||||
)
|
||||
|
||||
|
||||
def main():
|
||||
# Parse command line arguments
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("--correlation", "-c", type=np.float32, required=True)
|
||||
parser.add_argument("-N", type=np.int32, required=True)
|
||||
parser.add_argument("--plot", "-p", action="store_true")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# Generate & plot data
|
||||
|
||||
means = np.array([0, 0])
|
||||
cov = np.array([[1, args.correlation], [args.correlation, 1]])
|
||||
|
||||
x = np.random.multivariate_normal(means, cov, size=args.N)
|
||||
|
||||
print(twodim_array_to_pgfplots_table_string(x))
|
||||
|
||||
if args.plot:
|
||||
plt.scatter(x[:, 0], x[:, 1])
|
||||
plt.show()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@ -1,40 +0,0 @@
|
||||
import argparse
|
||||
import numpy as np
|
||||
import matplotlib.pyplot as plt
|
||||
from scipy.special import binom
|
||||
|
||||
|
||||
def array_to_pgfplots_table_string(a):
|
||||
return " ".join([f"({k}, {val})" for (k, val) in enumerate(a)]) + f" ({len(a)}, 0)"
|
||||
|
||||
|
||||
def P_binom(N, p, k):
|
||||
return binom(N, k) * p**k * (1 - p) ** (N - k)
|
||||
|
||||
|
||||
def main():
|
||||
# Parse command line arguments
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("-N", type=np.int32, required=True)
|
||||
parser.add_argument("-p", type=np.float32, required=True)
|
||||
parser.add_argument("--show", "-s", action="store_true")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# Generate and show data
|
||||
|
||||
N = args.N
|
||||
p = args.p
|
||||
|
||||
bars = np.array([P_binom(N, p, k) for k in range(N + 1)])
|
||||
|
||||
print(array_to_pgfplots_table_string(bars))
|
||||
|
||||
if args.show:
|
||||
plt.stem(bars)
|
||||
plt.show()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue
Block a user