check_columns.py

#!/usr/bin/env python3
"""Check TEPR/STRR columns"""

import pandas as pd
from utils.excel_loader import load_submarine_forecast_data

data = load_submarine_forecast_data('./dataset/submarine_forecast.xlsx')

print("TEPR columns:", data['tepr'].columns.tolist())
print("TEPR sample:")
print(data['tepr'].head())
print()

print("STRR columns:", data['strr'].columns.tolist())
print("STRR sample:")
print(data['strr'].head())