如何更简洁实现列表所有元素组合的编码表格生成？

方案1：利用itertools.product生成二进制掩码 #

每个元素的选择状态可直接对应0（不选）或1（选），所有组合本质就是n个元素的0/1笛卡尔积，这种方式逻辑直观且无需额外补值操作：

import pandas as pd
from itertools import product

sample_list = ['a', 'b', 'c', 'd']
n = len(sample_list)

# 生成所有0/1组合，直接对应每个元素的选择状态
mask = list(product([0, 1], repeat=n))
df = pd.DataFrame(mask, columns=sample_list)

方案2：利用numpy向量化操作生成掩码 #

当列表元素较多时，numpy的向量操作效率优势更明显，通过整数转二进制的方式生成完整掩码矩阵：

import pandas as pd
import numpy as np

sample_list = ['a', 'b', 'c', 'd']
n = len(sample_list)

# 生成0到2^n-1的整数，转二进制后补全前导零，再转为整数矩阵
nums = np.arange(2**n)
mask = np.array([list(f"{num:0{n}b}") for num in nums], dtype=int)
df = pd.DataFrame(mask, columns=sample_list)

对比原方案的优势 #

原方案需要遍历不同长度的组合、多次创建DataFrame后拼接，还需fillna补0，步骤繁琐且中间冗余数据多；新方案直接生成完整的0/1掩码矩阵，一步构建DataFrame，逻辑更清晰，执行效率更高，代码量也大幅减少。

内容的提问来源于stack exchange，提问作者Quixotic22