from pathlib import Path
import pandas as pd
import numpy as np
from _shared import load
ROOT = Path(__file__).resolve().parents[1]
OUT = ROOT / "outputs"
OUT.mkdir(exist_ok=True)
orders = load("orders").copy()
mask = orders["status"].isin(["paid","refunded"])
selected = orders.loc[mask,["order_id","status","amount"]].copy()
orders.loc[mask,"paid_flag"] = True
orders.loc[~mask,"paid_flag"] = False
assert len(selected) == int(mask.sum())
assert orders.loc[mask,"paid_flag"].eq(True).all()
a = pd.Series([10,20],index=["u1","u2"])
b = pd.Series([1,2],index=["u2","u1"])
aligned = a+b
assert aligned.to_dict() == {"u1":12,"u2":21}
print("标签对齐:",aligned.to_dict())
print("位置相加:",(a.to_numpy()+b.to_numpy()).tolist())
print(selected.head().to_string(index=False))
assert len(orders.iloc[:0]) == 0