BigCodeBench / BigCodeBench v0.1.0_hf 179985e5-c722-59ed-a227-7a210fcdacaf

complete prompt

import numpy as np
from scipy.stats import ttest_1samp
import matplotlib.pyplot as plt

# Constants
ALPHA = 0.05


def task_func(data_matrix):
    """
    Calculate the mean value of each row in a 2D data matrix, run a t-test from a sample against the population value, and record the mean values that differ significantly.
    - Create a lineplot with the mean of rows in red. Its label is 'Means'.
    - Create a line plot with the significant_indices (those with a pvalue less than ALPHA) on the x-axis and the corresponding means on the y-axis. This plot should be blue. Its label is 'Significant Means'.
    - Create an horizontal line which represent the mean computed on the whole 2D matrix. It should be in green. Its label is 'Population Mean'.

    Parameters:
    data_matrix (numpy.array): The 2D data matrix.

    Returns:
    tuple: A tuple containing:
        - list: A list of indices of the means that are significantly different from the population mean.
        - Axes: The plot showing the means and significant means.

    Requirements:
    - numpy
    - scipy.stats.ttest_1samp
    - matplotlib.pyplot

    Example:
    >>> data = np.array([[6, 8, 1, 3, 4], [-1, 0, 3, 5, 1]])
    >>> indices, ax = task_func(data)
    >>> print(indices)
    []

    Example 2:
    >>> data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
    >>> indices, ax = task_func(data)
    >>> print(indices)
    []
    """

instruct prompt

Calculate the mean value of each row in a 2D data matrix, run a t-test from a sample against the population value, and record the mean values that differ significantly. - Create a lineplot with the mean of rows in red. Its label is 'Means'. - Create a line plot with the significant_indices (those with a pvalue less than ALPHA) on the x-axis and the corresponding means on the y-axis. This plot should be blue. Its label is 'Significant Means'. - Create an horizontal line which represent the mean computed on the whole 2D matrix. It should be in green. Its label is 'Population Mean'. Example 2: >>> data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) >>> indices, ax = task_func(data) >>> print(indices) []
The function should output with:
    tuple: A tuple containing:
    list: A list of indices of the means that are significantly different from the population mean.
    Axes: The plot showing the means and significant means.
You should write self-contained code starting with:

Code

import numpy as np
from scipy.stats import ttest_1samp
import matplotlib.pyplot as plt
# Constants
ALPHA = 0.05
def task_func(data_matrix):

code prompt

Code

import numpy as np
from scipy.stats import ttest_1samp
import matplotlib.pyplot as plt
# Constants
ALPHA = 0.05
def task_func(data_matrix):

entry point

task_func

libs

• numpy
• matplotlib
• scipy