如何生成包含col_1-col_5指定-1组合的测试DataFrame

列取值规则 #

• col_1 至 col_5：可从各自对应的有效值列表中采样取值，或用-1表示任意值
• col_6 至 col_8：只能从各自对应的有效值列表中随机采样取值

核心生成要求 #

需生成满足以下条件的测试DataFrame：

1. 覆盖col_1至col_5的所有指定-1组合（组合列表见下方）
2. 每个组合对应的行规则：
   • 元组中列出的列固定设为-1
   • 其余col_1至col_5列从对应有效值列表中随机采样取值
   • col_6至col_8列全程从对应有效值列表中随机采样取值

组合示例说明 #

示例1：2列非-1、2列-1的情况：对应行内，2列是有效值列表的随机采样值，2列固定为-1，剩余1列也为随机有效值，col_6-col_8均为随机有效值
示例2：2列非-1、3列-1的情况：对应行内，2列是随机有效值，3列固定为-1，col_6-col_8均为随机有效值

指定的-1组合列表 #

[(),
 ('col_1',),
 ('col_2',),
 ('col_3',),
 ('col_4',),
 ('col_5',),
 ('col_1', 'col_2'),
 ('col_1', 'col_3'),
 ('col_1', 'col_4'),
 ('col_1', 'col_5'),
 ('col_2', 'col_3'),
 ('col_2', 'col_4'),
 ('col_2', 'col_5'),
 ('col_3', 'col_4'),
 ('col_3', 'col_5'),
 ('col_4', 'col_5'),
 ('col_1', 'col_2', 'col_3'),
 ('col_1', 'col_2', 'col_4'),
 ('col_1', 'col_2', 'col_5'),
 ('col_1', 'col_3', 'col_4'),
 ('col_1', 'col_3', 'col_5'),
 ('col_1', 'col_4', 'col_5'),
 ('col_2', 'col_3', 'col_4'),
 ('col_2', 'col_3', 'col_5'),
 ('col_2', 'col_4', 'col_5'),
 ('col_3', 'col_4', 'col_5'),
 ('col_1', 'col_2', 'col_3', 'col_4'),
 ('col_1', 'col_2', 'col_3', 'col_5'),
 ('col_1', 'col_2', 'col_4', 'col_5'),
 ('col_1', 'col_3', 'col_4', 'col_5'),
 ('col_2', 'col_3', 'col_4', 'col_5'),
 ('col_1', 'col_2', 'col_3', 'col_4', 'col_5')]

内容的提问来源于stack exchange，提问作者Karthik K V