python typedef

from typing import List, Tuple

Vector = List[float]
Matrix = List[List[float]]
RGB = Tuple[int, int, int]

def add(a: int, b: int) -> int:
    return a + b

from typing import List, Tuple

# 定义类型别名
Vector = List[float]
Matrix = List[List[float]]
RGB = Tuple[int, int, int]

# 使用类型注解的函数
def scale(scalar: float, vector: Vector) -> Vector:
    return [scalar * num for num in vector]

def get_average_color(rgb_image: List[RGB]) -> RGB:
    total_color = [sum(col) for col in zip(*rgb_image)]
    return tuple(c // len(rgb_image) for c in total_color)

# 示例使用
v = Vector([1.0, 2.0, 3.0])
scaled_v = scale(2.0, v)
print(scaled_v)  # 输出: [2.0, 4.0, 6.0]

image = [
    RGB(255, 0, 0),
    RGB(0, 255, 0),
    RGB(0, 0, 255)
]
avg_color = get_average_color(image)
print(avg_color)  # 输出: (85, 85, 85)