A distribuição amostral

import numpy as np
import matplotlib.pyplot as plt

size = int(input())

# Não altere as linhas abaixo: o tamanho em MB de cada arquivo na unidade
rng = np.random.default_rng(21)
files = np.round(rng.random(6000) ** 4 * 300 + 5, 2)


# Sua função de Sample Means Vary: as médias de count amostras
def sample_means(data, size, count):
    means = []
    for i in range(count):
        sample = rng.choice(data, size=size, replace=False)
        means.append(sample.mean())
    return np.array(means)


fig, axs = plt.subplots(1, 2)
axs[0].set_title("All Files")
axs[0].set_xlabel("File size (MB)")
axs[1].set_title("Sample Means")
axs[1].set_xlabel("Mean file size (MB)")

# Escreva seu código abaixo