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

Question

大家早上好

我试图从 itertools.combination 函数生成的元组列表中插入索引 [0] 和索引 1 到变量 x 和 y 中

在这些链接'itertools.combinations' object is not subscriptable我发现组合中的对象不可下标，并且想知道是否有办法将单独的元组索引获取到函数中

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 返回一个生成器表达式。

如果您想访问combinations 结果的索引，您必须使用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)