import numpy as np
from PIL import Image
import matplotlib.pyplot as plt

# 합성 위성 사진 4장 (4 카테고리)
np.random.seed(42)
city  = np.random.randint(80,160,(128,128,3),np.uint8)
farm  = np.array([[[80,140,60]]128]128,np.uint8)
ocean = np.array([[[30,80,160]]128]128,np.uint8)
forest= np.array([[[30,100,40]]128]128,np.uint8)

print(city.shape)  # (128, 128, 3)
print(city.dtype)  # uint8 (0~255)

# R, G, B 채널만 따로
fig, ax = plt.subplots(1, 3, figsize=(12,4))
ax[0].imshow(city[:,:,0], cmap="Reds")
ax[1].imshow(city[:,:,1], cmap="Greens")
ax[2].imshow(city[:,:,2], cmap="Blues")
plt.show()

# 카테고리별 평균 RGB
for name, img in [("city",city),("farm",farm),
                  ("ocean",ocean),("forest",forest)]:
    r,g,b = img[:,:,0].mean(), img[:,:,1].mean(), img[:,:,2].mean()
    print(f"{name}: R={r:.0f} G={g:.0f} B={b:.0f}")