[Python] 型ヒント”union”の使い方

from typing import Union
def process_data(data: Union[int, str]) -> None:
    # データの型に応じて処理を行う
    if isinstance(data, int):
        print(f"整数データ: {data}")
    elif isinstance(data, str):
        print(f"文字列データ: {data}")
# 関数の呼び出し例
process_data(10)
process_data("こんにちは")

整数データ: 10
文字列データ: こんにちは

from typing import Union, List, Tuple
def analyze_data(data: Union[int, str, List[int], Tuple[str, ...]]) -> None:
    # データの型に応じて処理を行う
    if isinstance(data, int):
        print(f"整数データ: {data}")
    elif isinstance(data, str):
        print(f"文字列データ: {data}")
    elif isinstance(data, list):
        print(f"整数のリスト: {data}")
    elif isinstance(data, tuple):
        print(f"文字列のタプル: {data}")
# 関数の呼び出し例
analyze_data(42)
analyze_data("Python")
analyze_data([1, 2, 3])
analyze_data(("a", "b", "c"))

整数データ: 42
文字列データ: Python
整数のリスト: [1, 2, 3]
文字列のタプル: ('a', 'b', 'c')

from typing import Optional
def greet_user(name: Optional[str] = None) -> str:
    # 名前が指定されていない場合の処理
    if name is None:
        return "こんにちは、ゲストさん！"
    else:
        return f"こんにちは、{name}さん！"
# 関数の呼び出し例
print(greet_user())
print(greet_user("太郎"))

こんにちは、ゲストさん！
こんにちは、太郎さん！

from typing import Union
def calculate_area(shape: Union[int, float, str]) -> float:
    # shapeが整数または浮動小数点数の場合、正方形の面積を計算
    if isinstance(shape, (int, float)):
        return shape * shape
    # shapeが文字列の場合、円の面積を計算
    elif isinstance(shape, str):
        radius = float(shape)
        return 3.14159 * radius * radius
    else:
        raise ValueError("無効な型です")
# 関数の呼び出し例
print(calculate_area(5))       # 正方形の面積
print(calculate_area("3.5"))   # 円の面積

25.0
38.484475

from typing import Union
def parse_input(data: str) -> Union[int, float, str]:
    # 入力データを整数に変換できるか試みる
    try:
        return int(data)
    except ValueError:
        pass
    # 入力データを浮動小数点数に変換できるか試みる
    try:
        return float(data)
    except ValueError:
        pass
    # 変換できない場合はそのまま文字列として返す
    return data
# 関数の呼び出し例
print(parse_input("42"))      # 整数として返す
print(parse_input("3.14"))    # 浮動小数点数として返す
print(parse_input("Python"))  # 文字列として返す

42
3.14
Python

from typing import Union
class Config:
    # 設定値は整数または文字列を許容
    setting: Union[int, str]
    def __init__(self, setting: Union[int, str]) -> None:
        self.setting = setting
    def display_setting(self) -> None:
        print(f"現在の設定値: {self.setting}")
# クラスのインスタンス化例
config1 = Config(10)
config2 = Config("デフォルト")
config1.display_setting()
config2.display_setting()

現在の設定値: 10
現在の設定値: デフォルト

from typing import Union, List
def process_dataset(dataset: List[Union[int, float, str]]) -> None:
    for data in dataset:
        if isinstance(data, int):
            print(f"整数データ: {data}")
        elif isinstance(data, float):
            print(f"浮動小数点データ: {data}")
        elif isinstance(data, str):
            print(f"文字列データ: {data}")
# データセットの例
dataset = [42, 3.14, "Python", 100, "データ解析"]
process_dataset(dataset)

整数データ: 42
浮動小数点データ: 3.14
文字列データ: Python
整数データ: 100
文字列データ: データ解析

from typing import Union
def handle_user_input(input_data: Union[int, str]) -> str:
    if isinstance(input_data, int):
        return f"数値入力を受け付けました: {input_data}"
    elif isinstance(input_data, str):
        return f"文字列入力を受け付けました: {input_data}"
# ユーザー入力の例
print(handle_user_input(123))
print(handle_user_input("こんにちは"))

数値入力を受け付けました: 123
文字列入力を受け付けました: こんにちは

from typing import Union, List
def train_model(data: List[Union[int, float]], epochs: Union[int, str]) -> None:
    if isinstance(epochs, str):
        epochs = int(epochs)  # 文字列を整数に変換
    print(f"モデルを{epochs}エポックでトレーニングします。")
    # モデルのトレーニング処理(省略)
# モデルのトレーニング例
train_model([0.1, 0.2, 0.3], 10)
train_model([0.1, 0.2, 0.3], "20")

モデルを10エポックでトレーニングします。
モデルを20エポックでトレーニングします。