Python：如何重复数据框中1到n行的每个组合？

Question

已获得如下所示的数据帧df：

Store   Aisle   Table
11      59      2
11      61      3

需要将行的每个组合展开3次，生成新的列“位”，范围值如下：

Store   Aisle   Table   Bit
11      59      2       1
11      59      2       2
11      59      2       3
11      61      3       1
11      61      3       2
11      61      3       3

尝试了以下代码，但没有成功。

df.loc[df.index.repeat(range(3))]

帮帮我！提前谢谢。

Answer 1

我发现直接使用concat可以：

new = pd.concat([df]*3).sort_index().reset_index(drop=True)

>>> print(new)

   Store  Aisle  Table
0     11     59      2
1     11     59      2
2     11     59      2
3     11     61      3
4     11     61      3
5     11     61      3

然后，您可以稍后添加值为1 2 3的新列，因为情况总是这样（从这里开始）：

new['Bit'] =  np.r_[:len(new)] % 3 + 1

Answer 2

df = df.iloc[np.repeat(np.arange(len(df)), 3)]
df['Bit'] = list(range(1, len(df)//3+1))*3

Answer 3

您应该提供一个数字，而不是一个重复的范围。此外，您还需要一点处理：

(df.loc[df.index.repeat(3)]
   .assign(Bit=lambda d: d.groupby(level=0).cumcount().add(1))
   .reset_index(drop=True)
)

输出：

   Store  Aisle  Table  Bit
0     11     59      2    1
1     11     59      2    2
2     11     59      2    3
3     11     61      3    1
4     11     61      3    2
5     11     61      3    3

或者，使用多索引。从\u product：

idx = pd.MultiIndex.from_product([df.index, range(1,3+1)], names=(None, 'Bit'))
(df.reindex(idx.get_level_values(0))
   .assign(Bit=idx.get_level_values(1))
)