从 itertool.combination 列表中获取元组索引

Question

大家早上好

我正在尝试从 itertools.combination 函数

生成的元组列表中插入变量 x 和 y 的索引 [0] 和索引

在这些 link 中，我发现组合中的对象不可订阅，想知道是否有办法将单独的元组索引获取到函数中

a = [10,11,12,13,14,15]
b = combinations(a,2)

x= b[0]
y= b[1]

conditions = [df[f'sma_{x}'] > df[f'sma_{y}'],
              df[f'sma_{x}'] < df[f'sma_{y}']]

choices = [1, -1]

df[f'sma_{x} & sma_{y}'] = np.select(conditions, choices, default=0)

Answer 1

itertools.combinations return 生成器表达式。

如果您想访问 combinations 结果的索引，您必须 'consume' 使用 list:

a = [10,11,12,13,14,15]
b = list(combinations(a,2))

x= b[0]
y= b[1]

conditions = [df[f'sma_{x}'] > df[f'sma_{y}'],
              df[f'sma_{x}'] < df[f'sma_{y}']]

choices = [1, -1]

df[f'sma_{x} & sma_{y}'] = np.select(conditions, choices, default=0)

从 itertool.combination 列表中获取元组索引

Get Tuple Index from itertool.combination list

python

combinations

tuples

list

itertools