используй _ вместо i для переменных, которые не нужны

[total_weight + 1 for _ in range(n)]