benchmarks.wiki / Public workspace

BigCodeBench / BigCodeBench v0.1.0_hf 26c94cd5-59e1-5985-9f4a-4003768e6180

Problem

Answer published by the source. Consult the official source to check your work against its answer.

complete prompt

import os
import random
import pandas as pd


def task_func(data_dir,
          csv_files=['file1.csv', 'file2.csv', 'file3.csv'],
          seed=None):
    """
    Randomly select one of the provided csv_files and select a certain number 
    of records from the file at random.
    The selected records are returned in a DataFrame. 
    The name of the selected csv_file is also returned.

    If the csv_file is empty return an empty DataFrame.

    Parameters:
    data_dir (str): The directory where the CSV files are located.
    csv_files (list of str): The list of CSV files to choose from. Default is ['file1.csv', 'file2.csv', 'file3.csv'].
    seed (int, optional): Seed for random number generation and for sampling from the csv.
    
    Returns:
    tuple: A tuple containing two elements:
        - str: The name of the randomly selected file.
        - DataFrame: A pandas DataFrame with the selected rows.

    Requirements:
    - os
    - random
    - pandas

    Example:
    >>> file_name, df = task_func('test_data')
    >>> print(file_name)
    'file2.csv'
    >>> print(df)
           Animal     Weight
     0        Cat          1
    21      Mouse         12
    15   Elephant       1000
     2      Tiger        500
    """

instruct prompt

Randomly select one of the provided csv_files and select a certain number of records from the file at random. The selected records are returned in a DataFrame. The name of the selected csv_file is also returned. If the csv_file is empty return an empty DataFrame.
The function should output with:
    tuple: A tuple containing two elements:
    str: The name of the randomly selected file.
    DataFrame: A pandas DataFrame with the selected rows.
You should write self-contained code starting with:

Code

import os
import random
import pandas as pd
def task_func(data_dir,
          csv_files=['file1.csv', 'file2.csv', 'file3.csv'],
          seed=None):

code prompt

Code

import os
import random
import pandas as pd
def task_func(data_dir,
          csv_files=['file1.csv', 'file2.csv', 'file3.csv'],
          seed=None):

entry point

task_func

libs

  • pandas
  • random
  • os

Discussion

Discussion

No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

See answer Answer published by the source

Artifacts

Code, notes and reproducible work shared by participants. Files are served from a separate origin.

No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

Source and history

Official source

initial import