Python 코드 리팩터링

모듈성 개선을 위해 Python 코드를 리팩터링하고 기능을 추가합니다.

모델을 직접 쿼리하고 Cloud 콘솔에서 다른 매개변수 값을 사용하거나 Vertex AI API를 직접 호출하여 반환된 결과를 테스트할 수 있습니다.

Freeform 프롬프트

임의 이름을 생성하고 콘솔의 메뉴에서 사용자가 선택한 항목에 따라 다양한 종류의 데이터 처리를 수행하는 Python 프로그램이 있습니다. data_processor 함수를 분할하고 완전히 삭제하여 코드를 리팩터링하도록 도와줄 수 있나요? 또한 역방향 약어 이름에 대한 기능을 추가하고 사용자 입력 검색을 별도의 함수로 이동해 주세요. 기본 가드에는 함수 호출 하나만 포함되어야 합니다.

import random
from faker import Faker

def transform_data(data):
    upper_names = [name.upper() for name in data]
    lower_names = [name.lower() for name in data]
    reversed_names = [name[::-1] for name in data]
    acronym_names = ["".join([word[0] for word in name.split()]) for name in data]
    return upper_names, lower_names, reversed_names, acronym_names

def calculate_statistics(data):
    name_lengths = [len(name) for name in data]
    average_length = sum(name_lengths) / len(name_lengths)
    max_length = max(name_lengths)
    min_length = min(name_lengths)
    return average_length, max_length, min_length

def group_by_first_letter(data):
    grouped_data = {}
    for name in data:
        grouped_data.setdefault(name[0], []).append(name)
    return grouped_data

def analyze_name_frequencies(data):
    name_counts = {}
    for name in data:
        name_counts[name] = name_counts.get(name, 0) + 1
    return name_counts

def display_menu():
    print("Menu:")
    print("1. Display original names")
    print("2. Display uppercased names")
    print("3. Display lowercased names")
    print("4. Display reversed names")
    print("5. Display acronym names")
    print("6. Display name statistics")
    print("7. Display names grouped by first letter")
    print("8. Display name frequencies")
    print("9. Exit")
    choice = input("Enter your choice: ")
    return choice

if __name__ == "__main__":
    # Generate random names
    fake = Faker()
    names = [fake.name() for _ in range(10)]

    upper_names, lower_names, reversed_names, acronym_names = transform_data(names)
    avg_length, max_length, min_length = calculate_statistics(names)
    grouped_names = group_by_first_letter(names)
    name_frequencies = analyze_name_frequencies(names)

    processed_data = {
        "original_names": names,
        "upper_names": upper_names,
        "lower_names": lower_names,
        "reversed_names": reversed_names,
        "acronym_names": acronym_names,
        "average_length": avg_length,
        "max_length": max_length,
        "min_length": min_length,
        "grouped_names": grouped_names,
        "name_frequencies": name_frequencies
    }

    while True:
        choice = display_menu()
        if choice == "1":
            print(processed_data["original_names"])
        elif choice == "2":
            print(processed_data["upper_names"])
        elif choice == "3":
            print(processed_data["lower_names"])
        elif choice == "4":
            print(processed_data["reversed_names"])
        elif choice == "5":
            print(processed_data["acronym_names"])
        elif choice == "6":
            print(f"Average name length: {processed_data['average_length']}")
            print(f"Maximum name length: {processed_data['max_length']}")
            print(f"Minimum name length: {processed_data['min_length']}")
        elif choice == "7":
            for letter, names in processed_data["grouped_names"].items():
                print(f"Names starting with {letter}: {names}")
        elif choice == "8":
            for name, count in processed_data["name_frequencies"].items():
                print(f"{name}: {count}")
        elif choice == "9":
            break
        else:
            print("Invalid choice")

응답

import random
from faker import Faker

def generate_names(num_names=10):
    fake = Faker()
    return [fake.name() for _ in range(num_names)]

def transform_names(names):
    upper_names = [name.upper() for name in names]
    lower_names = [name.lower() for name in names]
    reversed_names = [name[::-1] for name in names]
    acronym_names = ["".join(word[0] for word in name.split()) for name in names]
    reversed_acronym_names = ["".join(word[0] for word in name.split())[::-1] for name in names] #Added reversed acronym
    return upper_names, lower_names, reversed_names, acronym_names, reversed_acronym_names

def calculate_statistics(names):
    name_lengths = [len(name) for name in names]
    average_length = sum(name_lengths) / len(name_lengths) if name_lengths else 0
    max_length = max(name_lengths) if name_lengths else 0
    min_length = min(name_lengths) if name_lengths else 0
    return average_length, max_length, min_length

def group_by_first_letter(names):
    grouped_data = {}
    for name in names:
        grouped_data.setdefault(name[0], []).append(name)
    return grouped_data

def analyze_name_frequencies(names):
    name_counts = {}
    for name in names:
        name_counts[name] = name_counts.get(name, 0) + 1
    return name_counts

def get_user_input():
    choice = input("Enter your choice: ")
    return choice

def display_menu():
    print("Menu:")
    print("1. Display original names")
    print("2. Display uppercased names")
    print("3. Display lowercased names")
    print("4. Display reversed names")
    print("5. Display acronym names")
    print("6. Display reversed acronym names") #Added menu option
    print("7. Display name statistics")
    print("8. Display names grouped by first letter")
    print("9. Display name frequencies")
    print("10. Exit")
    return get_user_input()

def main():
    names = generate_names()
    upper_names, lower_names, reversed_names, acronym_names, reversed_acronym_names = transform_names(names)
    avg_length, max_length, min_length = calculate_statistics(names)
    grouped_names = group_by_first_letter(names)
    name_frequencies = analyze_name_frequencies(names)

    processed_data = {
        "original_names": names,
        "upper_names": upper_names,
        "lower_names": lower_names,
        "reversed_names": reversed_names,
        "acronym_names": acronym_names,
        "reversed_acronym_names": reversed_acronym_names,
        "average_length": avg_length,
        "max_length": max_length,
        "min_length": min_length,
        "grouped_names": grouped_names,
        "name_frequencies": name_frequencies
    }

    while True:
        choice = display_menu()
        try:
            choice = int(choice)
            if choice == 1:
                print(processed_data["original_names"])
            elif choice == 2:
                print(processed_data["upper_names"])
            elif choice == 3:
                print(processed_data["lower_names"])
            elif choice == 4:
                print(processed_data["reversed_names"])
            elif choice == 5:
                print(processed_data["acronym_names"])
            elif choice == 6:
                print(processed_data["reversed_acronym_names"])
            elif choice == 7:
                print(f"Average name length: {processed_data['average_length']}")
                print(f"Maximum name length: {processed_data['max_length']}")
                print(f"Minimum name length: {processed_data['min_length']}")
            elif choice == 8:
                for letter, names in processed_data["grouped_names"].items():
                    print(f"Names starting with {letter}: {names}")
            elif choice == 9:
                for name, count in processed_data["name_frequencies"].items():
                    print(f"{name}: {count}")
            elif choice == 10:
                break
            else:
                print("Invalid choice")
        except ValueError:
            print("Invalid input. Please enter a number.")


if __name__ == "__main__":
    main()
모델: gemini-1.5-flash-002
강도: 0.2
최대 출력 토큰: 8192
최상위 K: 40.0
최상위 P: 0.95