Polars vs Pandas năm 2026: Hiệu năng, Cú pháp và Câu hỏi Phỏng vấn Data Analyst

# benchmark_comparison.py
import polars as pl
import pandas as pd
import time

# Polars: lazy evaluation với predicate pushdown
start = time.perf_counter()
result_polars = (
    pl.scan_parquet("sales_data_14gb.parquet")  # Lazy: chưa tải dữ liệu
    .filter(pl.col("region") == "EMEA")         # Predicate được đẩy xuống file reader
    .group_by("product_category")
    .agg(pl.col("revenue").sum())
    .collect()                                   # Thực thi xảy ra ở đây
)
polars_time = time.perf_counter() - start

# Pandas: eager evaluation tải toàn bộ file
start = time.perf_counter()
df = pd.read_parquet("sales_data_14gb.parquet")  # Toàn bộ 14 GB được tải vào bộ nhớ
result_pandas = (
    df[df["region"] == "EMEA"]                   # Filter áp dụng sau khi tải
    .groupby("product_category")["revenue"]
    .sum()
)
pandas_time = time.perf_counter() - start

print(f"Polars: {polars_time:.2f}s | Pandas: {pandas_time:.2f}s")
# Output điển hình: Polars: 1.2s | Pandas: 13.2s

# lazy_evaluation_example.py
import polars as pl

# Định nghĩa lazy query (chưa thực thi)
lazy_query = (
    pl.scan_csv("transactions_50gb.csv")   # LazyFrame: chỉ schema, không có dữ liệu
    .filter(pl.col("amount") > 1000)       # Thêm vào query plan
    .filter(pl.col("status") == "completed")  # Kết hợp với filter trên
    .select(["transaction_id", "amount", "customer_id"])  # Projection pushdown
    .group_by("customer_id")
    .agg([
        pl.col("amount").sum().alias("total_spent"),
        pl.col("transaction_id").count().alias("transaction_count")
    ])
)

# Xem query plan đã tối ưu
print(lazy_query.explain())
# Hiển thị: predicate pushdown, projection pushdown, filter combination

# Thực thi khi sẵn sàng
result = lazy_query.collect()

# pandas_3_new_features.py
import pandas as pd

# Backend chuỗi PyArrow giờ là mặc định (thao tác chuỗi nhanh hơn 5-10x)
df = pd.read_csv("users.csv")  # Chuỗi giờ là string[pyarrow] mặc định

# Expression builder mới (cú pháp tương tự Polars)
result = df.select(
    pd.col("name").str.upper(),
    pd.col("age") * 2,
    (pd.col("salary") > 100000).alias("high_earner")
)

# Copy-on-Write loại bỏ SettingWithCopyWarning
subset = df[df["age"] > 30]  # Trả về view, không phải copy
subset = subset.copy()        # Copy tường minh cần thiết cho mutation

# Các phương thức interop Arrow mới
import pyarrow as pa
arrow_table = pa.table({"x": [1, 2, 3]})
df = pd.DataFrame.from_arrow(arrow_table)  # Import zero-copy

# filtering_comparison.py
import polars as pl
import pandas as pd

# Dữ liệu mẫu
data = {
    "name": ["Alice", "Bob", "Charlie", "Diana"],
    "department": ["Engineering", "Sales", "Engineering", "HR"],
    "salary": [95000, 72000, 88000, 65000],
    "years": [5, 3, 7, 2]
}

# POLARS: Tham chiếu cột tường minh với pl.col()
df_pl = pl.DataFrame(data)
result_pl = (
    df_pl
    .filter(pl.col("department") == "Engineering")  # Filter với pl.col()
    .filter(pl.col("salary") > 80000)               # Chain filters
    .select(["name", "salary"])                     # Chọn cột
)

# PANDAS: Ký hiệu bracket
df_pd = pd.DataFrame(data)
result_pd = (
    df_pd
    .loc[df_pd["department"] == "Engineering"]  # loc cho filtering
    .loc[lambda x: x["salary"] > 80000]         # Lambda cho chaining
    [["name", "salary"]]                        # Bracket cho selection
)

# aggregation_comparison.py

# POLARS: Cú pháp aggregation biểu cảm
result_pl = (
    df_pl
    .group_by("department")
    .agg([
        pl.col("salary").mean().alias("avg_salary"),
        pl.col("salary").max().alias("max_salary"),
        pl.col("name").count().alias("headcount"),
        (pl.col("salary") * pl.col("years")).sum().alias("total_compensation_years")
    ])
)

# PANDAS: Cú pháp named aggregation
result_pd = (
    df_pd
    .groupby("department")
    .agg(
        avg_salary=("salary", "mean"),
        max_salary=("salary", "max"),
        headcount=("name", "count"),
        total_compensation_years=("salary", lambda x: (df_pd.loc[x.index, "salary"] * df_pd.loc[x.index, "years"]).sum())
    )
)

# joins_comparison.py

employees = pl.DataFrame({
    "emp_id": [1, 2, 3],
    "name": ["Alice", "Bob", "Charlie"],
    "dept_id": [10, 20, 10]
})

departments = pl.DataFrame({
    "dept_id": [10, 20, 30],
    "dept_name": ["Engineering", "Sales", "HR"]
})

# POLARS: Cú pháp join tường minh
result_pl = employees.join(
    departments,
    on="dept_id",      # Cột join
    how="left"         # Loại join: left, inner, outer, cross, semi, anti
)

# PANDAS: Hàm merge
result_pd = pd.merge(
    employees.to_pandas(),
    departments.to_pandas(),
    on="dept_id",
    how="left"
)

# hybrid_workflow.py
import polars as pl
from sklearn.ensemble import RandomForestClassifier
import matplotlib.pyplot as plt

# Xử lý dữ liệu nặng với Polars (nhanh hơn 10x)
df = (
    pl.scan_parquet("raw_data/*.parquet")
    .filter(pl.col("valid") == True)
    .with_columns([
        (pl.col("revenue") / pl.col("quantity")).alias("unit_price"),
        pl.col("timestamp").dt.month().alias("month")
    ])
    .group_by(["customer_id", "month"])
    .agg([
        pl.col("revenue").sum(),
        pl.col("quantity").mean()
    ])
    .collect()
)

# Chuyển đổi sang Pandas cho ML (zero-copy cho cột số)
X = df.select(["revenue", "quantity"]).to_pandas()
y = df.select("churn").to_pandas().values.ravel()

# scikit-learn mong đợi Pandas/NumPy
model = RandomForestClassifier()
model.fit(X, y)

# Vẽ đồ thị với tích hợp Pandas
df.to_pandas().plot(kind="bar", x="month", y="revenue")
plt.savefig("monthly_revenue.png")

# interview_answer_large_file.py
import polars as pl

# Tùy chọn 1: Lazy evaluation với streaming (Polars)
result = (
    pl.scan_csv("large_file.csv")  # Chỉ đọc schema
    .filter(pl.col("status") == "active")
    .group_by("region")
    .agg(pl.col("revenue").sum())
    .collect(streaming=True)  # Xử lý theo batch
)

# Tùy chọn 2: Chunked processing (fallback Pandas)
import pandas as pd

results = []
for chunk in pd.read_csv("large_file.csv", chunksize=1_000_000):
    filtered = chunk[chunk["status"] == "active"]
    agg = filtered.groupby("region")["revenue"].sum()
    results.append(agg)

final = pd.concat(results).groupby(level=0).sum()

# interview_conversion.py

# Mã Pandas cho trước
df = pd.read_csv("sales.csv")
df["year"] = pd.to_datetime(df["date"]).dt.year
result = (
    df[df["amount"] > 1000]
    .groupby(["region", "year"])
    .agg({"amount": ["sum", "mean"], "customer_id": "nunique"})
)

# Tương đương Polars
result = (
    pl.scan_csv("sales.csv")  # Lazy để tối ưu hóa
    .with_columns(
        pl.col("date").str.to_datetime().dt.year().alias("year")
    )
    .filter(pl.col("amount") > 1000)
    .group_by(["region", "year"])
    .agg([
        pl.col("amount").sum().alias("amount_sum"),
        pl.col("amount").mean().alias("amount_mean"),
        pl.col("customer_id").n_unique().alias("unique_customers")
    ])
    .collect()
)

# migration_strategy.py
import polars as pl
import pandas as pd

# Bước 1: Giữ Pandas cho khám phá nhanh
def explore_data(path: str) -> pd.DataFrame:
    return pd.read_csv(path).head(1000)

# Bước 2: Giới thiệu Polars cho các phép biến đổi nặng
def process_data(path: str) -> pl.DataFrame:
    return (
        pl.scan_csv(path)
        .filter(pl.col("valid") == True)
        .with_columns([
            (pl.col("price") * pl.col("quantity")).alias("total")
        ])
        .collect()
    )

# Bước 3: Chuyển đổi ở ranh giới khi cần
def train_model(df_polars: pl.DataFrame):
    df_pandas = df_polars.to_pandas()  # Zero-copy cho số
    # mã scikit-learn ở đây

# Bước 4: Thay thế hot paths dần dần
# Xác định các thao tác Pandas chậm bằng profiling
# Thay thế bằng tương đương Polars từng hàm một